Unverified Commit e1039a1c authored by Umang Yadav's avatar Umang Yadav Committed by GitHub
Browse files

Remove use of int8x4 format for the rocblas (#1920)

* do not use int8x4 format for the rocblas
parent 77c05035
...@@ -55,9 +55,16 @@ bool get_compute_fp32_flag() ...@@ -55,9 +55,16 @@ bool get_compute_fp32_flag()
bool get_int8_x4_format(context& ctx) bool get_int8_x4_format(context& ctx)
{ {
#if ROCBLAS_VERSION_MAJOR >= 3
(void)(ctx);
return false;
#else
// int8x4 packed format is only available starting from rocblas-v2.38 and it is deprecated in
// v3.0 and will be removed in v4.0
rocblas_gemm_flags flag; rocblas_gemm_flags flag;
rocblas_query_int8_layout_flag(ctx.get_stream().get_rocblas(), &flag); rocblas_query_int8_layout_flag(ctx.get_stream().get_rocblas(), &flag);
return flag == rocblas_gemm_flags_pack_int8x4; return flag == rocblas_gemm_flags_pack_int8x4;
#endif
} }
} // namespace gpu } // namespace gpu
} // namespace MIGRAPHX_INLINE_NS } // namespace MIGRAPHX_INLINE_NS
......
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