Bug rocblas int8 (#767)

* fix the flag in rocblas api for int8 data type * used different flag for different rocblas versions * clang format Co-authored-by: mvermeulen <[email protected]>
ROCm · Mar 10, 2021 · 63e04f5 · 63e04f5
1 parent 197b3a4
commit 63e04f5
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/src/targets/gpu/gemm_impl.cpp b/src/targets/gpu/gemm_impl.cpp
@@ -108,7 +108,11 @@ void gemm_impl(
                            compute_type,
                            rocblas_gemm_algo_standard,
                            0,
+#if ROCBLAS_VERSION_MAJOR >= 2 && ROCBLAS_VERSION_MINOR >= 38
+                           rocblas_gemm_flags_pack_int8x4);
+#else
                            0);
+#endif
         }
         else
         {
@@ -141,7 +145,11 @@ void gemm_impl(
                            compute_type,
                            rocblas_gemm_algo_standard,
                            0,
+#if ROCBLAS_VERSION_MAJOR >= 2 && ROCBLAS_VERSION_MINOR >= 38
+                           rocblas_gemm_flags_pack_int8x4);
+#else
                            0);
+#endif
         }
     });
 }