Skip to content

Commit 936d47b

Browse files
authored
Minor rocblas related changes for rocm 70 ROCm/xla#206 (#2985)
1 parent eaa2aa5 commit 936d47b

File tree

2 files changed

+2
-3
lines changed

2 files changed

+2
-3
lines changed

third_party/xla/third_party/gpus/rocm_configure.bzl

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -736,7 +736,7 @@ def _create_local_rocm_repository(repository_ctx):
736736
"%{hipruntime_version_number}": rocm_config.hipruntime_version_number,
737737
"%{hipblaslt_flag}": have_hipblaslt,
738738
"%{hip_soversion_number}": "6" if int(rocm_config.rocm_version_number) >= 60000 else "5",
739-
"%{rocblas_soversion_number}": "4" if int(rocm_config.rocm_version_number) >= 60000 else "3",
739+
"%{rocblas_soversion_number}": "5" if int(rocm_config.rocm_version_number) >= 70000 else "4",
740740
},
741741
)
742742

@@ -755,7 +755,7 @@ def _create_local_rocm_repository(repository_ctx):
755755
"%{hipruntime_version_number}": rocm_config.hipruntime_version_number,
756756
"%{hipblaslt_flag}": have_hipblaslt,
757757
"%{hip_soversion_number}": "6" if int(rocm_config.rocm_version_number) >= 60000 else "5",
758-
"%{rocblas_soversion_number}": "4" if int(rocm_config.rocm_version_number) >= 60000 else "3",
758+
"%{rocblas_soversion_number}": "5" if int(rocm_config.rocm_version_number) >= 70000 else "4",
759759
},
760760
)
761761

third_party/xla/xla/stream_executor/rocm/rocblas_wrapper.h

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -263,7 +263,6 @@ using tsl::internal::CachedDsoLoader::GetRocblasDsoHandle;
263263
__macro(rocblas_gemm_batched_ex_get_solutions_by_type) \
264264
__macro(rocblas_gemm_strided_batched_ex_get_solutions) \
265265
__macro(rocblas_is_managing_device_memory) \
266-
__macro(rocblas_is_user_managing_device_memory) \
267266
__macro(rocblas_set_workspace) \
268267
__macro(rocblas_strsm_batched) \
269268
__macro(rocblas_dtrsm_batched) \

0 commit comments

Comments
 (0)