composable_kernel

mirror of https://github.com/ROCm/composable_kernel.git synced 2026-05-14 10:09:41 +00:00

Files

rocking5566 5cdbf1273a Int8 qunatization gemm xdl (#108 )

* Add int8 of mk_nk_mn to the ckProfiler

* Add example of int8 gemm

* Fix typo, use ushort instead of half_t for bfloat16

* replace ushortXXX_t to bhalfXXX_t

* rename ushort to bhalf_t

* Add bf16 example

* Add bf16 gemm to ckProfiler

* Fix alignment

* Fix typo

* Add unit test for gemm_xdl int8

* Add gemm_xdl fp32 unit test

* Add gemm_xdl bf16 unit test

* fix build

* fix build issue due to merge conflict

* Fix build

* Fix build error

* [What] gemm + relu inference
[How] gemm + requant + relu + requant + clamp

* clean

Co-authored-by: rocking <chunylai@amd.com>
Co-authored-by: Chao Liu <chao.liu2@amd.com>

[ROCm/composable_kernel commit: ad41aa0e7a]

2022-03-05 00:48:09 -06:00

1_gemm_xdl

Int8 qunatization gemm xdl (#108 )

2022-03-05 00:48:09 -06:00

2_gemm_xdl_bias_relu

GEMM+Bias+ReLU+Add (#76 )

2022-02-06 22:32:47 -06:00

3_gemm_xdl_bias_relu_add

GEMM+Bias+ReLU+Add (#76 )

2022-02-06 22:32:47 -06:00

4_conv2d_fwd_xdl

Add gridwise GEMM pipeline (#89 )