mirror of https://github.com/ROCm/composable_kernel.git synced 2026-04-19 22:39:03 +00:00

Files

Adam Osewski 061ac0649c Polished Grouped GEMM APIs and new BF16 instances (#1600 )

* Few small fixes.

* New GroupedGemm instances (BF16)

* Unify and refactor GroupedGEMM device API.

* Adapt changes to new API.

* Adapt grouped gemm profiler.

* Accept multiple kbatches for grouped gemm profiler.

- delete obsolete two stage as it is now covered by grouped gemm

* Update unit test for grouped gemm.

* Fix thresholds for BF16 and F8. Unblock tests.

* Fix few instances.

* Multiple small fixes.

* Adapt to new API, check dynamic casting.

* Uncomment few data types in grouped gemm profiler.

* Fix call to SetDeviceArgs.

* Fix profile grouped gemm multiply tile loop.

* Fix grouped gemm tile loop kernel args in client examples.

* Review comments.

2024-11-27 13:02:44 +01:00

CMakeLists.txt

Grouped GEMM Multiple D tile loop. (#1247 )

2024-04-25 15:12:53 -05:00

grouped_gemm_multiple_d_dl_fp16.cpp

update copyright headers (#726 )

2023-05-31 18:46:57 -05:00

grouped_gemm_multiple_d_splitk_xdl_fp16.cpp

Polished Grouped GEMM APIs and new BF16 instances (#1600 )

2024-11-27 13:02:44 +01:00

grouped_gemm_multiple_d_xdl_fp16.cpp

Polished Grouped GEMM APIs and new BF16 instances (#1600 )

2024-11-27 13:02:44 +01:00

grouped_gemm_xdl_bf16.cpp

Refactoring cmake files to build data types separately. (#932 )

2023-09-20 22:15:56 -07:00

grouped_gemm_xdl_fixed_nk_bias_fp16.cpp

Polished Grouped GEMM APIs and new BF16 instances (#1600 )

2024-11-27 13:02:44 +01:00

grouped_gemm_xdl_fixed_nk_fp16_fp8.cpp

Polished Grouped GEMM APIs and new BF16 instances (#1600 )

2024-11-27 13:02:44 +01:00

grouped_gemm_xdl_fixed_nk_fp16.cpp

Polished Grouped GEMM APIs and new BF16 instances (#1600 )

2024-11-27 13:02:44 +01:00

grouped_gemm_xdl_fp16.cpp

update copyright headers (#726 )

2023-05-31 18:46:57 -05:00

grouped_gemm_xdl_fp32.cpp

update copyright headers (#726 )

2023-05-31 18:46:57 -05:00

grouped_gemm_xdl_int4.cpp

update copyright headers (#726 )

2023-05-31 18:46:57 -05:00

grouped_gemm_xdl_int8.cpp

update copyright headers (#726 )

2023-05-31 18:46:57 -05:00

grouped_gemm_xdl_splitk_fp16.cpp

changed test for grouped_gemm to be random (#959 )

2023-10-03 09:32:58 -05:00

README.md

Code clean-up (#1285 )

2024-05-10 09:41:39 -07:00

run_grouped_gemm_example.inc

Polished Grouped GEMM APIs and new BF16 instances (#1600 )

2024-11-27 13:02:44 +01:00

README.md

Instructions for `example_grouped_gemm_xdl`

Run `example_grouped_gemm_xdl`

#arg1: verification (0=no, 1=yes)
#arg2: initialization (0=no init, 1=integer value, 2=decimal value)
#arg3: run kernel # of times (>1)
./bin/example_grouped_gemm_xdl_fp16 0 1 5

README.md

Instructions for example_grouped_gemm_xdl

Run example_grouped_gemm_xdl

Instructions for `example_grouped_gemm_xdl`

Run `example_grouped_gemm_xdl`