Add rocBLAS to the project

build_tools/fetch_sources.py

@@ -132,6 +132,7 @@ def main(argv):
             # Math Libraries
             "hipBLAS-common",
             "hipBLASLt",
+            "rocBLAS",
         ],
     )
     args = parser.parse_args(argv)

math-libs/CMakeLists.txt

@@ -61,4 +61,41 @@ therock_cmake_subproject_glob_c_sources(hipBLASLt
   SUBDIRS
     .
 )
+therock_cmake_subproject_provide_package(hipBLASLt hipblaslt lib/cmake/hipblaslt)
 therock_cmake_subproject_activate(hipBLASLt)
+
+#################################################################################
+# rocBLAS
+#################################################################################
+
+therock_cmake_subproject_declare(rocBLAS
+  EXTERNAL_SOURCE_DIR "rocBLAS"
+  CMAKE_ARGS
+    -DHIP_PLATFORM=amd
+    -DROCM_PATH=
+    -DROCM_DIR=
+    # TODO: Fix passing TENSILE_ROCM_ASSEMBLER_PATH.
+    #       Enable building with hipBLASLt afterwards.
+    -DBUILD_WITH_TENSILE=OFF
+    -DBUILD_WITH_HIPBLASLT=OFF
+    # -DTensile_LOGIC=
+    # -DTensile_CODE_OBJECT_VERSION=default
+    # -DTensile_CPU_THREADS=
+    # -DTensile_LIBRARY_FORMAT=msgpack
+    # "-DTensile_ROCM_ASSEMBLER_PATH=${_toolchain_dir}/bin/clang++"
+    # "-DTensile_ROCM_OFFLOAD_BUNDLER_PATH=${_toolchain_dir}/bin/clang-offload-bundler"
+    # TODO: Enable clients.
+  IGNORE_PACKAGES
+    ROCM
+  COMPILER_TOOLCHAIN
+    amd-hip
+  RUNTIME_DEPS
+    #hipBLAS-common
+    #hipBLASLt
+    hip-clr
+)
+therock_cmake_subproject_glob_c_sources(rocBLAS
+  SUBDIRS
+    .
+)
+therock_cmake_subproject_activate(rocBLAS)

math-libs/post_hook_rocBLAS.cmake

@@ -0,0 +1,9 @@
+list(APPEND CMAKE_MODULE_PATH "${THEROCK_SOURCE_DIR}/cmake")
+include(therock_rpath)
+
+therock_set_install_rpath(
+  TARGETS
+    rocblas
+  PATHS
+    .
+)

math-libs/rocBLAS

@@ -0,0 +1 @@
+../sources/rocBLAS