From ca1ddefa3a4a4ccfb53682f9281542c0f76436c8 Mon Sep 17 00:00:00 2001 From: Jason Furmanek Date: Tue, 22 Oct 2024 14:35:12 +0000 Subject: [PATCH] [ROCm] Added include of hipblas.h in hipblaslt_wrapper.h --- xla/stream_executor/rocm/hipblaslt_wrapper.h | 1 + 1 file changed, 1 insertion(+) diff --git a/xla/stream_executor/rocm/hipblaslt_wrapper.h b/xla/stream_executor/rocm/hipblaslt_wrapper.h index c53cff6a93391..a17998a0bc72b 100644 --- a/xla/stream_executor/rocm/hipblaslt_wrapper.h +++ b/xla/stream_executor/rocm/hipblaslt_wrapper.h @@ -22,6 +22,7 @@ limitations under the License. #include "rocm/rocm_config.h" #if TF_HIPBLASLT +#include "rocm/include/hipblas/hipblas.h" #if TF_ROCM_VERSION >= 50500 #include "rocm/include/hipblaslt/hipblaslt.h" #else