composable_kernel

mirror of https://github.com/ROCm/composable_kernel.git synced 2026-05-14 02:02:46 +00:00

Files

rocking5566 6822598f8a Support alpha beta scaling for GEMM (#78 )

* [What] Add 2d version of bias, prepare to implement alpha / beta scaling

* Add alpha / beta functor

* Refine parameter of example

* [What] Use real type instead of template
[Why] Prevent implicit cast

* Rename parameter for general operator

* Remove redundant comment

* Fix compile error

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

[ROCm/composable_kernel commit: 6f928a0876]

2022-02-11 00:48:41 -06:00

1_gemm_xdl

GEMM+Bias+ReLU+Add (#76 )

2022-02-06 22:32:47 -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

GEMM+Bias+ReLU+Add (#76 )