diff --git a/third_party/gpus/rocm_configure.bzl b/third_party/gpus/rocm_configure.bzl index 0fd9a5af240808..91ec090d5fd45d 100644 --- a/third_party/gpus/rocm_configure.bzl +++ b/third_party/gpus/rocm_configure.bzl @@ -742,7 +742,7 @@ def _create_local_rocm_repository(repository_ctx): "%{hipruntime_version_number}": rocm_config.hipruntime_version_number, "%{hipblaslt_flag}": have_hipblaslt, "%{hip_soversion_number}": "6" if int(rocm_config.rocm_version_number) >= 60000 else "5", - "%{rocblas_soversion_number}": "4" if int(rocm_config.rocm_version_number) >= 60000 else "3", + "%{rocblas_soversion_number}": "5" if int(rocm_config.rocm_version_number) >= 60500 else "4", }, ) @@ -761,7 +761,7 @@ def _create_local_rocm_repository(repository_ctx): "%{hipruntime_version_number}": rocm_config.hipruntime_version_number, "%{hipblaslt_flag}": have_hipblaslt, "%{hip_soversion_number}": "6" if int(rocm_config.rocm_version_number) >= 60000 else "5", - "%{rocblas_soversion_number}": "4" if int(rocm_config.rocm_version_number) >= 60000 else "3", + "%{rocblas_soversion_number}": "5" if int(rocm_config.rocm_version_number) >= 60500 else "4", }, ) diff --git a/third_party/xla/third_party/gpus/rocm_configure.bzl b/third_party/xla/third_party/gpus/rocm_configure.bzl index c02479d1adad4c..fc8b79ae5f6dc2 100644 --- a/third_party/xla/third_party/gpus/rocm_configure.bzl +++ b/third_party/xla/third_party/gpus/rocm_configure.bzl @@ -736,7 +736,7 @@ def _create_local_rocm_repository(repository_ctx): "%{hipruntime_version_number}": rocm_config.hipruntime_version_number, "%{hipblaslt_flag}": have_hipblaslt, "%{hip_soversion_number}": "6" if int(rocm_config.rocm_version_number) >= 60000 else "5", - "%{rocblas_soversion_number}": "4" if int(rocm_config.rocm_version_number) >= 60000 else "3", + "%{rocblas_soversion_number}": "5" if int(rocm_config.rocm_version_number) >= 60500 else "4", }, ) @@ -755,7 +755,7 @@ def _create_local_rocm_repository(repository_ctx): "%{hipruntime_version_number}": rocm_config.hipruntime_version_number, "%{hipblaslt_flag}": have_hipblaslt, "%{hip_soversion_number}": "6" if int(rocm_config.rocm_version_number) >= 60000 else "5", - "%{rocblas_soversion_number}": "4" if int(rocm_config.rocm_version_number) >= 60000 else "3", + "%{rocblas_soversion_number}": "5" if int(rocm_config.rocm_version_number) >= 60500 else "4", }, )