diff --git a/xla/stream_executor/rocm/hipblaslt_wrapper.h b/xla/stream_executor/rocm/hipblaslt_wrapper.h
index c53cff6a933913..7d8268c8c5d2fa 100644
--- a/xla/stream_executor/rocm/hipblaslt_wrapper.h
+++ b/xla/stream_executor/rocm/hipblaslt_wrapper.h
@@ -23,6 +23,7 @@ limitations under the License.
 
 #if TF_HIPBLASLT
 #if TF_ROCM_VERSION >= 50500
+#include "rocm/include/hipblas/hipblas.h"
 #include "rocm/include/hipblaslt/hipblaslt.h"
 #else
 #include "rocm/include/hipblaslt.h"