Unverified Commit 63e04f59 authored by Shucai Xiao's avatar Shucai Xiao Committed by GitHub
Browse files

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: default avatarmvermeulen <5479696+mvermeulen@users.noreply.github.com>
parent 197b3a46
......@@ -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
}
});
}
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment