build(rocm): support rocm-3.9

GitOrigin-RevId: 85f8911736fae8c5e9b0a71629c89e78029e8c55

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")
 

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>