Use rocblas_INCLUDE_DIR to define path to librocblas

Author: nilsfriess

cmake/FindrocBLAS.cmake

@@ -51,11 +51,18 @@ find_package_handle_standard_args(rocBLAS
     VERSION_VAR
       ROCBLAS_VERSION
 )
+
+if (DEFINED rocblas_INCLUDE_DIR)
+  set(ROCBLAS_LIB_PATH "${rocblas_INCLUDE_DIR}/../lib/librocblas.so")
+else()
+  set(ROCBLAS_LIB_PATH "${HIP_PATH}/../rocblas/lib/librocblas.so")
+endif()
+
 # OPENCL_INCLUDE_DIR
 if(NOT TARGET ONEMKL::rocBLAS::rocBLAS)
   add_library(ONEMKL::rocBLAS::rocBLAS SHARED IMPORTED)
   set_target_properties(ONEMKL::rocBLAS::rocBLAS PROPERTIES
-      IMPORTED_LOCATION "${HIP_PATH}/../rocblas/lib/librocblas.so"
+      IMPORTED_LOCATION "${rocblas_INCLUDE_DIR}/../lib/librocblas.so"
       INTERFACE_INCLUDE_DIRECTORIES "${ROCBLAS_INCLUDE_DIR};${HIP_INCLUDE_DIRS};"
       INTERFACE_LINK_LIBRARIES "Threads::Threads;${ROCBLAS_LIBRARIES};"
   )