composable_kernel

mirror of https://github.com/ROCm/composable_kernel.git synced 2026-04-19 14:29:05 +00:00

Files

ruanjm 04dd314883 [CK_TILE] Add Various Fusion Functions to RMSNorm (#1802 )

* Add shortcut to RMSNorm

* Modify test for adding shortcut for RMSNorm

* Add fused parameter into tests

* 1. Add YDataType. 2. rmsnorm2d_fwd_traits_ from rmsnorm2d_fwd.hpp to rmsnorm2d_fwd_api.cpp and rmsnorm2d_fwd_instance_common.hpp

* 1. Supports various stride and percisions.

* Add support of Epilogue

* Add fuse and epilogue support to rmsnorm ref

* Modify rmsnorm example

* Refactor tests/examples

* Bug fix for newly added tests/examples

* Bug fix for new tests 2

* Modify smoke test scripts

remove dbg code

* Supports non-smooth dyanmic quant

* Update Rmsnorm2dFwd::GetName()

* rename xscale and prec_sx to smoothscale and prec_sm

Bug fix after rename

Remove files

* change example_rmsnorm2d_fwd.cpp

* update performance calculator

* Fix issue in two-pass when fuse add is enabled

* Remove comment of beta

---------

Co-authored-by: rocking <ChunYu.Lai@amd.com>

2025-01-15 10:23:48 +08:00

reference_batched_dropout.hpp

CK Tile FA Training kernels (#1286 )

2024-06-04 13:12:45 -05:00

reference_batched_elementwise.hpp

introducing ck_tile! (#1216 )

2024-04-15 19:27:12 -05:00

reference_batched_gemm.hpp

introducing ck_tile! (#1216 )

2024-04-15 19:27:12 -05:00

reference_batched_masking.hpp

introducing ck_tile! (#1216 )