Added rocblas header (microsoft#4538)

Co-authored-by: Logan Adams <114770087+loadams@users.noreply.github.com>
mauryaavinash95 · Feb 17, 2024 · 31f73c5 · 31f73c5
1 parent 680eb52
commit 31f73c5
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/csrc/includes/gemm_test.h b/csrc/includes/gemm_test.h
@@ -9,6 +9,9 @@
 #ifndef __HIP_PLATFORM_HCC__
 #include <cuda_profiler_api.h>
 #endif
+#ifdef __HIP_PLATFORM_HCC__
+#include <rocblas/rocblas.h>
+#endif
 #include <array>
 #include <cstdio>
 #include <cstdlib>