Add missing RocBlas include (microsoft#4557)

mauryaavinash95 · Feb 17, 2024 · d9fb529 · d9fb529
1 parent b34bbd3
commit d9fb529
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,