Added rocblas header

microsoft · Oct 18, 2023 · e641c3a · e641c3a
1 parent a735881
commit e641c3a
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>