composable_kernel icon indicating copy to clipboard operation
composable_kernel copied to clipboard

GEMM specification not found in CK?

Open chengl-amd opened this issue 2 years ago • 1 comments

Hi there,

I am wondering which impl or example I can refer to if I want to run:

RELU(GEMM (A[M,K]_INT8, B[K,N]_INT8) + bias_INT8) = output_INT8.

I tried example/03_gemm_bias_relu, but it fails to build when I set:

using ADataType = int8_t; using BDataType = int8_t; using AccDataType = F32; using CShuffleDataType = F16; using DDataType = int8_t; using EDataType = int8_t;

KR

chengl-amd avatar Nov 10 '23 10:11 chengl-amd

Hi @chengl-amd , try this branch , I changed AccDataType and CShuffleDataType to int32_t (which is supported with int8_t).

bartekxk avatar Jan 15 '24 13:01 bartekxk

@chengl-amd Is your issue resolved? If so, please close the ticket. Thanks!

ppanchad-amd avatar Aug 21 '24 17:08 ppanchad-amd

@chengl-amd Closing ticket. Please re-open the ticket if you still see the issue with the latest ROCm. Thanks!

ppanchad-amd avatar Sep 25 '24 18:09 ppanchad-amd