ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-06-30 11:47:48 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
40ed20a30d043dc98ec426fc247a2aaa30d079bf
composable_kernel/include/ck/tensor_operation/gpu/block
History
mtgu0705 40ed20a30d updated the codes
2025-06-04 02:04:28 -05:00
..
blockwise_gemm_dl_v2r3.hpp
Implement DPP8 based GEMM for Navi21 (#826)
2023-08-14 15:46:27 -05:00
blockwise_gemm_dlops_v2r2.hpp
update copyright headers (#726)
2023-05-31 18:46:57 -05:00
blockwise_gemm_dlops_v3.hpp
update copyright headers (#726)
2023-05-31 18:46:57 -05:00
blockwise_gemm_dpp.hpp
Fix cmake warnings (#1342)
2024-06-21 09:47:58 +02:00
blockwise_gemm_mx_pipeline_xdlops_base.hpp
Use packed_size_v for A/BPackedSize
2025-06-03 06:10:07 +00:00
blockwise_gemm_pipeline_wmma_selector.hpp
DeviceGemm_Wmma_CShuffleV3 with BlockGemmPipelineVersion::v3 (#2096)
2025-04-28 10:14:21 +05:00
blockwise_gemm_pipeline_wmmaops_base.hpp
DeviceGemm_Wmma_CShuffleV3 with BlockGemmPipelineVersion::v3 (#2096)
2025-04-28 10:14:21 +05:00
blockwise_gemm_pipeline_wmmaops_v3.hpp
DeviceGemm_Wmma_CShuffleV3 with BlockGemmPipelineVersion::v3 (#2096)
2025-04-28 10:14:21 +05:00
blockwise_gemm_pipeline_wmmaops.hpp
DeviceGemm_Wmma_CShuffleV3 with BlockGemmPipelineVersion::v3 (#2096)
2025-04-28 10:14:21 +05:00
blockwise_gemm_pipeline_xdlops_ab_scale_selector.hpp
[A8W8 GEMM] Optimized weight-preshuffled implementation & add quantization datatype for CK TILE rms_norm (#1862)
2025-02-20 14:00:27 -08:00
blockwise_gemm_pipeline_xdlops_b_preshuffle_dequant_v1.hpp
Ck int4 moe develop (#1949)
2025-03-10 11:16:44 +08:00
blockwise_gemm_pipeline_xdlops_b_preshuffle_dequant_v3.hpp
Refactor thread_copy_lds_direct_load; fix gfx942 direct lds load example; fix f16_pki4 example
2025-06-03 14:54:30 +00:00
blockwise_gemm_pipeline_xdlops_b_preshuffle_gufusion_dequant_v1.hpp
Enable splitk for mxfp4; clang format;
2025-06-02 12:23:01 +00:00
blockwise_gemm_pipeline_xdlops_b_preshuffle_gufusion_v1.hpp
Enable splitk for mxfp4; clang format;
2025-06-02 12:23:01 +00:00
blockwise_gemm_pipeline_xdlops_b_preshuffle_gufusion_v3.hpp
revert unexpected file change
2025-05-13 09:55:33 +08:00
blockwise_gemm_pipeline_xdlops_b_preshuffle_mx_moe_gufusion_v1.hpp
mx_moe_fp4 ready for aiter with clang-format.
2025-05-16 04:09:26 -05:00
blockwise_gemm_pipeline_xdlops_b_preshuffle_mx_moe_gufusion_v3.hpp
update code
2025-05-17 09:28:26 -05:00
blockwise_gemm_pipeline_xdlops_b_preshuffle_mx_moe_selector.hpp
Merge remote-tracking branch 'origin/feiw/mxfp4_moe_2Stages' into wip-f4
2025-05-28 09:28:26 +00:00
blockwise_gemm_pipeline_xdlops_b_preshuffle_mx_moe_v1.hpp
update code
2025-05-17 09:28:26 -05:00
blockwise_gemm_pipeline_xdlops_b_preshuffle_mx_moe_v3.hpp
commit with debug info
2025-05-19 21:47:35 -05:00
blockwise_gemm_pipeline_xdlops_b_preshuffle_selector.hpp
Clang-format after 2 merges
2025-05-28 11:16:00 +00:00
blockwise_gemm_pipeline_xdlops_b_preshuffle_v1.hpp
Merge remote-tracking branch 'origin/f8blk_scale_opt' into wip-f4-mergemoe-2
2025-05-28 11:15:22 +00:00
blockwise_gemm_pipeline_xdlops_b_preshuffle_v2.hpp
Merge remote-tracking branch 'origin/feiw/mxfp4_moe_2Stages' into wip-f4
2025-05-28 09:28:26 +00:00
blockwise_gemm_pipeline_xdlops_b_preshuffle_v3.hpp
Merge remote-tracking branch 'origin/wip-f4-wp' into wip-f4
2025-05-27 03:36:38 +00:00
blockwise_gemm_pipeline_xdlops_b_scale_selector.hpp
[A8W8 GEMM] Optimized weight-preshuffled implementation & add quantization datatype for CK TILE rms_norm (#1862)
2025-02-20 14:00:27 -08:00
blockwise_gemm_pipeline_xdlops_base.hpp
doc the kGroup definition
2025-06-03 08:53:04 +00:00
blockwise_gemm_pipeline_xdlops_blockscale_b_preshuffle_selector.hpp
update moe v1 pipeline
2025-05-13 08:42:49 +00:00
blockwise_gemm_pipeline_xdlops_blockscale_b_preshuffle_v1.hpp
Remove debug infos; Enable flags for blockscale f8
2025-05-29 08:21:54 +00:00
blockwise_gemm_pipeline_xdlops_blockscale_b_preshuffle_v3.hpp
Merge remote-tracking branch 'origin/moe_bs_fp8_no_asm' into gfx950-mxfp4
2025-05-30 03:15:47 +00:00
blockwise_gemm_pipeline_xdlops_moe_blockscale_b_preshuffle_gufusion_v1.hpp
Moe blockscale gemm1&gemm2 asm support for aiter. Suppression cmkae flag til new compler.
2025-05-17 04:33:52 -05:00
blockwise_gemm_pipeline_xdlops_moe_blockscale_b_preshuffle_gufusion_v3.hpp
Enable splitk for mxfp4; clang format;
2025-06-02 12:23:01 +00:00
blockwise_gemm_pipeline_xdlops_moe_blockscale_b_preshuffle_selector.hpp
Enable splitk for mxfp4; clang format;
2025-06-02 12:23:01 +00:00
blockwise_gemm_pipeline_xdlops_moe_blockscale_b_preshuffle_v1.hpp
Moe blockscale gemm1&gemm2 asm support for aiter. Suppression cmkae flag til new compler.
2025-05-17 04:33:52 -05:00
blockwise_gemm_pipeline_xdlops_moe_blockscale_b_preshuffle_v3.hpp
Enable splitk for mxfp4; clang format;
2025-06-02 12:23:01 +00:00
blockwise_gemm_pipeline_xdlops_mx_bpreshuffle_selector.hpp
Merge gemm_mx_common.hpp
2025-05-27 06:13:03 +00:00
blockwise_gemm_pipeline_xdlops_mx_moe_nbs_gufusion_v3.hpp
Fix warning
2025-06-03 06:48:49 +00:00
blockwise_gemm_pipeline_xdlops_mx_moe_nbs_selector.hpp
add pipeline v1 for MOE Gemm2
2025-05-30 05:25:43 +00:00
blockwise_gemm_pipeline_xdlops_mx_moe_nbs_v1.hpp
add pipeline v1 for MOE Gemm2
2025-05-30 05:25:43 +00:00
blockwise_gemm_pipeline_xdlops_mx_moe_nbs_v3.hpp
updated the codes
2025-06-04 02:04:28 -05:00
blockwise_gemm_pipeline_xdlops_mx_selector.hpp
Merge gemm_mx_common.hpp
2025-05-27 06:13:03 +00:00
blockwise_gemm_pipeline_xdlops_selector.hpp
[A8W8 GEMM] Optimized weight-preshuffled implementation & add quantization datatype for CK TILE rms_norm (#1862)
2025-02-20 14:00:27 -08:00
blockwise_gemm_pipeline_xdlops_v1_ab_scale.hpp
update moe v1 pipeline
2025-05-13 08:42:49 +00:00
blockwise_gemm_pipeline_xdlops_v1_b_scale.hpp
Implement the fp16xint4 scale weight only kernel for Ali (#1786)
2025-01-03 18:35:21 +08:00
blockwise_gemm_pipeline_xdlops_v1_mx.hpp
Fix v1; use M padding
2025-05-26 10:32:26 +00:00
blockwise_gemm_pipeline_xdlops_v1.hpp
Fix cmake warnings (#1342)
2024-06-21 09:47:58 +02:00
blockwise_gemm_pipeline_xdlops_v2_ab_scale.hpp
[Block Scale GEMM] Optimized block scale gemm (#1950)
2025-03-11 10:11:21 -07:00
blockwise_gemm_pipeline_xdlops_v2_b_scale.hpp
Implement the fp16xint4 scale weight only kernel for Ali (#1786)
2025-01-03 18:35:21 +08:00
blockwise_gemm_pipeline_xdlops_v2.hpp
Change block gemm pipeline local prefill loop order. (#1692)
2024-11-26 17:36:53 +01:00
blockwise_gemm_pipeline_xdlops_v3_ab_scale.hpp
update moe v1 pipeline
2025-05-13 08:42:49 +00:00
blockwise_gemm_pipeline_xdlops_v3_b_scale.hpp
update moe v1 pipeline
2025-05-13 08:42:49 +00:00
blockwise_gemm_pipeline_xdlops_v3_mx_bpreshuffle.hpp
Enable splitk for mxfp4; clang format;
2025-06-02 12:23:01 +00:00
blockwise_gemm_pipeline_xdlops_v3_mx.hpp
add vmcnt guard for async copy
2025-05-28 03:47:46 +00:00
blockwise_gemm_pipeline_xdlops_v3.hpp
update moe v1 pipeline
2025-05-13 08:42:49 +00:00
blockwise_gemm_pipeline_xdlops_v4_b_scale.hpp
Implement the fp16xint4 scale weight only kernel for Ali (#1786)
2025-01-03 18:35:21 +08:00
blockwise_gemm_pipeline_xdlops_v4.hpp
[A8W8 GEMM] Optimized weight-preshuffled implementation & add quantization datatype for CK TILE rms_norm (#1862)
2025-02-20 14:00:27 -08:00
blockwise_gemm_pipeline_xdlops_v5.hpp
update moe v1 pipeline
2025-05-13 08:42:49 +00:00
blockwise_gemm_pipeline_xdlops.hpp
Fix compilation errors with Clang20.0. (#1533)
2024-09-25 13:45:38 -07:00
blockwise_gemm_smfmac_xdlops.hpp
Added structural sparsity blockwise gemm (#1435)
2024-09-11 15:19:42 +02:00
blockwise_gemm_wmma.hpp
fix compilation errors for gfx12 with clang20 (#1606)
2024-10-28 19:02:48 -07:00
blockwise_gemm_xdlops_skip_b_lds.hpp
Fix cmake warnings (#1342)
2024-06-21 09:47:58 +02:00
blockwise_gemm_xdlops.hpp
Merging the gfx12 code into public repo. (#1362)
2024-06-27 00:33:34 -07:00
blockwise_softmax.hpp
[HotFix] add config and version files to pass on build info (#856)
2023-08-23 11:36:17 -07:00
blockwise_tensor_slice_transfer_v5r1.hpp
update copyright headers (#726)
2023-05-31 18:46:57 -05:00
blockwise_welford.hpp
Batchnorm splitk single kernel (#771)
2023-07-06 10:58:55 -05:00
reduction_functions_blockwise.hpp
Batchnorm splitk single kernel (#771)
2023-07-06 10:58:55 -05:00
thread_group_tensor_slice_transfer_direct_load.hpp
Refactor thread_copy_lds_direct_load; fix gfx942 direct lds load example; fix f16_pki4 example
2025-06-03 14:54:30 +00:00
thread_group_tensor_slice_transfer_gather_direct_load.hpp
updated the codes
2025-06-04 02:04:28 -05:00
thread_group_tensor_slice_transfer_v4r1_dequant.hpp
Navi3 rel (#1176)
2024-03-08 17:11:51 -08:00
thread_group_tensor_slice_transfer_v4r1_gather.hpp
Moe gemm activation (#2026)
2025-04-23 10:35:34 +08:00
thread_group_tensor_slice_transfer_v4r1.hpp
[A8W8 GEMM] Optimized weight-preshuffled implementation & add quantization datatype for CK TILE rms_norm (#1862)
2025-02-20 14:00:27 -08:00
thread_group_tensor_slice_transfer_v4r2.hpp
Add elementwise with dynamic vector dim (#1198)
2024-03-22 10:40:43 +01:00
thread_group_tensor_slice_transfer_v6r1.hpp
update copyright headers (#726)
2023-05-31 18:46:57 -05:00
thread_group_tensor_slice_transfer_v6r1r2.hpp
initial stream-k implementation with example (#699)
2023-07-26 14:18:15 -05:00
thread_group_tensor_slice_transfer_v6r2.hpp
update copyright headers (#726)
2023-05-31 18:46:57 -05:00
thread_group_tensor_slice_transfer_v6r3.hpp
update copyright headers (#726)
2023-05-31 18:46:57 -05:00
thread_group_tensor_slice_transfer_v7.hpp
update copyright headers (#726)
2023-05-31 18:46:57 -05:00
thread_group_tensor_slice_transfer_v7r2.hpp
Codegen hipRTC compilation (#1579)
2025-01-31 09:48:39 -08:00
thread_group_tensor_slice_transfer_v7r3_scatter.hpp
Fix do_weight in gemm1. Fix cshuffle_datatype. Clang-format
2025-05-28 18:29:06 +08:00
thread_group_tensor_slice_transfer_v7r3.hpp
add f8 gemm multiD with both row/col wise scale (#1300)
2024-05-28 12:04:22 -05:00