[ROCm] fix hipblas headers for rocm version 6.3

ROCm · Nov 26, 2024 · c65e770 · c65e770
1 parent 76da730
commit c65e770
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/xla/stream_executor/rocm/hipblaslt_wrapper.h b/xla/stream_executor/rocm/hipblaslt_wrapper.h
@@ -20,6 +20,9 @@ limitations under the License.
 #include "rocm/rocm_config.h"
 
 #if TF_HIPBLASLT
+#if TF_ROCM_VERSION >= 60300
+#include "rocm/include/hipblas/hipblas.h"
+#endif
 #if TF_ROCM_VERSION >= 50500
 #include "rocm/include/hipblaslt/hipblaslt.h"
 #else