Add missing RocBlas include (#4557)

microsoft · Oct 23, 2023 · 869629c · 869629c
1 parent a02de22
commit 869629c
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/csrc/includes/cublas_wrappers.h b/csrc/includes/cublas_wrappers.h
@@ -13,6 +13,9 @@
 #ifndef __HIP_PLATFORM_AMD__
 #include <mma.h>
 #endif
+#ifdef __HIP_PLATFORM_AMD__
+#include <rocblas/rocblas.h>
+#endif
 #include <stdio.h>
 
 int cublas_gemm_ex(cublasHandle_t handle,