From 0083f4c4f35ce62b9f204c7420a5b06347065636 Mon Sep 17 00:00:00 2001
From: Megvii Engine Team <megengine@megvii.com>
Date: Tue, 23 Mar 2021 16:01:32 +0800
Subject: [PATCH] build(rocm): support rocm-3.9

GitOrigin-RevId: 85f8911736fae8c5e9b0a71629c89e78029e8c55
---
 cmake/rocm.cmake      | 4 ++--
 dnn/src/rocm/handle.h | 1 -
 2 files changed, 2 insertions(+), 3 deletions(-)

diff --git a/cmake/rocm.cmake b/cmake/rocm.cmake
index 6a7cb008..3bd5897a 100644
--- a/cmake/rocm.cmake
+++ b/cmake/rocm.cmake
@@ -65,7 +65,7 @@ set(HIP_LIBRARY_DIR ${HIP_ROOT_DIR}/../lib)
 function(find_rocm_library name dirname include library)
     find_path(${name}_LIBRARY_DIR
         NAMES ${library}
-        HINTS "${HIP_ROOT_DIR}/../${dirname}"
+        HINTS "${${name}_ROOT_DIR}" "${HIP_ROOT_DIR}/../${dirname}"
         PATH_SUFFIXES lib lib/x86_64
         DOC "Path to ${name} library directory")
 
@@ -75,7 +75,7 @@ function(find_rocm_library name dirname include library)
 
     find_path(${name}_INCLUDE_DIR
         NAMES ${include}
-        HINTS "${HIP_ROOT_DIR}/../${dirname}"
+        HINTS "${${name}_ROOT_DIR}" "${HIP_ROOT_DIR}/../${dirname}"
         PATH_SUFFIXES include
         DOC "Path to ${name} include directory")
 
diff --git a/dnn/src/rocm/handle.h b/dnn/src/rocm/handle.h
index 12eed9a1..367e9687 100644
--- a/dnn/src/rocm/handle.h
+++ b/dnn/src/rocm/handle.h
@@ -18,7 +18,6 @@
 #include "src/common/utils.h"
 #include "src/rocm/miopen_with_check.h"
 
-#include <rocblas-types.h>
 #include <rocblas.h>
 #include <atomic>
 #include <mutex>