From 6751acab271ecb153ae1fb7f78521f601bb6c018 Mon Sep 17 00:00:00 2001 From: Jason Furmanek Date: Mon, 21 Oct 2024 22:55:53 +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 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"