composable_kernel

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

Files

dummycoderfe eec0fed606 Ck tile/moe sorting (#1624 )

* add moe_sorting & check ok

* fix comments & typo

* Run remod.py under include/ck_tile & example/ck_tile directories

* format codes

* fix output ci check bug

* fix moe sorting readme and error commit file

* use magiv div to accelerate compute

* add an loop unroll for moe lds ops

* add extblocksnel to set zeros for moebufs

* [Ck_tile] moe set zero run ok, add size check and fix ref check

* [Ck_tile]fix moe_sorting fuse set_zero remod

* [Ck_tile] change name style, fix zero buffer size err, change folder

* [Ck_tile] moe_sorting: fix name style

* [Ck_tile] moe_sorting, remove useless params in traits

* [Ck_tile] change outputtile cnt * unit_size; change output buf alloc

---------

Co-authored-by: dummycoderfe <noplydummmycoder@163.com>
Co-authored-by: Po Yen, Chen <PoYen.Chen@amd.com>
Co-authored-by: carlushuang <carlus.huang@amd.com>

[ROCm/composable_kernel commit: bec6fbc65f]

2024-11-09 17:57:27 +08:00

reference

Ck tile/moe sorting (#1624 )

2024-11-09 17:57:27 +08:00

arg_parser.hpp

[CK_TILE] Update example README files & fix script compatibility issue (#1548 )

2024-10-08 10:45:12 +08:00

check_err.hpp

layernorm2d forward (#1339 )

2024-06-24 08:45:52 +08:00

convolution_host_tensor_descriptor_helper.hpp

[CK_TILE] Image to Column kernel (#1532 )