ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-05-20 12:59:49 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
0f8b58ac886f56191139f544b11a44028a4be970
composable_kernel/example
History
Gino Lu 0f8b58ac88 sparse_attn: R25 Step 1 A1 — per-warp PV-skip (paper Algorithm 1) + V0 instantiation 
Preserve the R25 Step 1 "A1 / redesign D" state before redesigning toward "B"
(per-CTA PV-skip matching upstream shipped reference). This snapshot lets us
restore A1 if the B redesign fails.

A1 redesign D pipeline (per-warp, arithmetic-only PV-skip, wrapped in
`if constexpr (kEnablePVSkip)`):
  - include/ck_tile/ops/sparse_attn/pipeline/block_fmha_pipeline_qr_ks_vs_async_sparge.hpp
  - include/ck_tile/ops/sparse_attn/kernel/fmha_fwd_sparge_kernel.hpp

V0 instantiation wiring (per gino_tmp/R25/programmer/v0_instance/REPORT.md):
  - example/ck_tile/50_sparse_attn/codegen/ops/fmha_fwd_sparge.py
  - example/ck_tile/50_sparse_attn/fmha_fwd_trek.hpp
  - example/ck_tile/50_sparse_attn/sparge_blockmap_trek.hpp
  - example/ck_tile/50_sparse_attn/sparge_blockmap_inst.cpp
  - example/ck_tile/50_sparse_attn/codegen/cpp_symbol_map.py
  - example/ck_tile/50_sparse_attn/CMakeLists.txt
  - example/ck_tile/01_fmha/CMakeLists.txt
  - example/ck_tile/50_sparse_attn/test_sparge.cpp (-pv_skip_compile=0|1 CLI)

This commit excludes all *_REVIEW.{hpp,cpp} mirror files (left untracked) and
all build artefacts. _vsa.hpp / _jenga.hpp are not modified.

Tag: R25-step1-A1-paper-aligned points at this commit.
2026-05-18 06:13:38 -04:00
..
01_gemm
Padding support for wave transfer (#3537)
2026-01-26 12:57:09 -08:00
02_gemm_bilinear
[CI, CK examples] Disable time_kernel for CI tests and examples (#3464)
2026-01-07 16:30:57 +01:00
03_gemm_bias_relu
chore(copyright) update library wide CMakeLists.txt copyright header template (#3313)
2025-11-28 13:49:54 -08:00
04_gemm_add_add_fastgelu
chore(copyright) update library wide CMakeLists.txt copyright header template (#3313)
2025-11-28 13:49:54 -08:00
09_convnd_fwd
[CK] Integrate GPU reference into ckProfiler for convolutions (#3379)
2025-12-18 07:59:45 +01:00
10_convnd_fwd_multiple_d_multiple_reduce
chore(copyright) update library wide CMakeLists.txt copyright header template (#3313)
2025-11-28 13:49:54 -08:00
11_convnd_fwd_bias
[DOCS] Documentation Addition (Readme updates) (#2495)
2025-10-16 03:10:57 -07:00
12_reduce
[rocm-libraries] ROCm/rocm-libraries#5030 (commit 8e02a26)
2026-03-06 16:28:22 +00:00
13_pool2d_fwd
[CI, CK examples] Disable time_kernel for CI tests and examples (#3464)
2026-01-07 16:30:57 +01:00
14_gemm_quantization
[rocm-libraries] ROCm/rocm-libraries#5348 (commit 7b18234)
2026-03-12 08:48:36 +00:00
15_grouped_gemm
[rocm-libraries] ROCm/rocm-libraries#4340 (commit 70a312f)
2026-02-26 00:28:58 +00:00
16_gemm_multi_d_multi_reduces
[rocm-libraries] ROCm/rocm-libraries#5348 (commit 7b18234)
2026-03-12 08:48:36 +00:00
17_convnd_bwd_data
[CK] Integrate GPU reference into ckProfiler for convolutions (#3379)
2025-12-18 07:59:45 +01:00
18_batched_gemm_reduce
chore(copyright) update library wide CMakeLists.txt copyright header template (#3313)
2025-11-28 13:49:54 -08:00
19_binary_elementwise
chore(copyright) update library wide CMakeLists.txt copyright header template (#3313)
2025-11-28 13:49:54 -08:00
20_grouped_conv_bwd_weight
[rocm-libraries] ROCm/rocm-libraries#4872 (commit ca623f7)
2026-02-25 20:11:01 +00:00
21_gemm_layernorm
chore(copyright) update library wide CMakeLists.txt copyright header template (#3313)
2025-11-28 13:49:54 -08:00
22_cgemm
[CI, CK examples] Disable time_kernel for CI tests and examples (#3464)
2026-01-07 16:30:57 +01:00
23_softmax
[CI, CK examples] Disable time_kernel for CI tests and examples (#3464)
2026-01-07 16:30:57 +01:00
24_batched_gemm
[CI, CK examples] Disable time_kernel for CI tests and examples (#3464)
2026-01-07 16:30:57 +01:00
25_gemm_bias_e_permute
Implement batched gemm bias permute for RDNA4 (#3534)
2026-01-17 08:30:27 +01:00
26_contraction
[rocm-libraries] ROCm/rocm-libraries#6323 (commit a668483)
2026-04-11 10:01:30 +00:00
27_layernorm2d_fwd
chore(copyright) update library wide CMakeLists.txt copyright header template (#3313)
2025-11-28 13:49:54 -08:00
28_grouped_gemm_bias_e_permute
chore(copyright) update library wide CMakeLists.txt copyright header template (#3313)
2025-11-28 13:49:54 -08:00
29_batched_gemm_bias_e_permute
Implement batched gemm bias permute for RDNA4 (#3534)
2026-01-17 08:30:27 +01:00
30_grouped_conv_fwd_multiple_d
[CI, CK examples] Disable time_kernel for CI tests and examples (#3464)
2026-01-07 16:30:57 +01:00
31_batched_gemm_gemm
chore(copyright) update library wide CMakeLists.txt copyright header template (#3313)
2025-11-28 13:49:54 -08:00
32_batched_gemm_scale_softmax_gemm
chore(copyright) update library wide CMakeLists.txt copyright header template (#3313)
2025-11-28 13:49:54 -08:00
33_multiple_reduce
[CI, CK examples] Disable time_kernel for CI tests and examples (#3464)
2026-01-07 16:30:57 +01:00
34_batchnorm
chore(copyright) update library wide CMakeLists.txt copyright header template (#3313)
2025-11-28 13:49:54 -08:00
35_splitK_gemm
[rocm-libraries] ROCm/rocm-libraries#4762 (commit 5598eb5)
2026-02-20 22:41:34 +00:00
36_sparse_embedding
[CI, CK examples] Disable time_kernel for CI tests and examples (#3464)
2026-01-07 16:30:57 +01:00
37_batched_gemm_add_add_relu_gemm_add
Implement batched gemm add relu gemm add for rdna4 (#3391)
2026-01-20 13:06:59 -08:00
38_grouped_conv_bwd_data_multiple_d
Grouped convolution backward data WMMA v3 implementation (#3460)
2025-12-30 16:25:08 +01:00
39_permute
chore(copyright) update library wide CMakeLists.txt copyright header template (#3313)
2025-11-28 13:49:54 -08:00
40_conv2d_fwd_quantization
[CI, CK examples] Disable time_kernel for CI tests and examples (#3464)
2026-01-07 16:30:57 +01:00
41_grouped_conv_conv_fwd
chore(copyright) update library wide CMakeLists.txt copyright header template (#3313)
2025-11-28 13:49:54 -08:00
42_groupnorm_fwd
[CI, CK examples] Disable time_kernel for CI tests and examples (#3464)
2026-01-07 16:30:57 +01:00
43_splitk_gemm_bias_e_permute
chore(copyright) update library wide CMakeLists.txt copyright header template (#3313)
2025-11-28 13:49:54 -08:00
44_elementwise_permute
[CI, CK examples] Disable time_kernel for CI tests and examples (#3464)
2026-01-07 16:30:57 +01:00
45_elementwise_normalization
[CI, CK examples] Disable time_kernel for CI tests and examples (#3464)
2026-01-07 16:30:57 +01:00
46_gemm_add_multiply
chore(copyright) update library wide CMakeLists.txt copyright header template (#3313)
2025-11-28 13:49:54 -08:00
47_gemm_bias_softmax_gemm_permute
chore(copyright) update library wide CMakeLists.txt copyright header template (#3313)
2025-11-28 13:49:54 -08:00
48_pool3d_fwd
chore(copyright) update library wide CMakeLists.txt copyright header template (#3313)
2025-11-28 13:49:54 -08:00
49_maxpool2d_bwd
chore(copyright) update library wide CMakeLists.txt copyright header template (#3313)
2025-11-28 13:49:54 -08:00
50_put_element
chore(copyright) update library wide CMakeLists.txt copyright header template (#3313)
2025-11-28 13:49:54 -08:00
51_avgpool3d_bwd
chore(copyright) update library wide CMakeLists.txt copyright header template (#3313)
2025-11-28 13:49:54 -08:00
52_im2col_col2im
chore(copyright) update library wide CMakeLists.txt copyright header template (#3313)
2025-11-28 13:49:54 -08:00
53_layernorm2d_bwd
chore(copyright) update library wide CMakeLists.txt copyright header template (#3313)
2025-11-28 13:49:54 -08:00
54_groupnorm_bwd
chore(copyright) update library wide CMakeLists.txt copyright header template (#3313)
2025-11-28 13:49:54 -08:00
59_grouped_gemm_multi_ABD
[rocm-libraries] ROCm/rocm-libraries#4425 (commit 513cf9f)
2026-02-25 05:17:08 +00:00
60_gemm_multi_ABD
[rocm-libraries] ROCm/rocm-libraries#6302 (commit 8d419e8)
2026-04-10 15:18:02 +00:00
61_contraction_multi_ABD
chore(copyright) update library wide CMakeLists.txt copyright header template (#3313)
2025-11-28 13:49:54 -08:00
62_convnd_activ
Adding remaining conv, dynamic_op, and scaleadd_scaleadd_relu flavors for grouped conv fwd (#3529)
2026-01-30 17:02:14 +01:00
63_layernorm4d_fwd
chore(copyright) update library wide CMakeLists.txt copyright header template (#3313)
2025-11-28 13:49:54 -08:00
64_fpAintB_gemm
chore(copyright) update library wide CMakeLists.txt copyright header template (#3313)
2025-11-28 13:49:54 -08:00
65_gemm_multiply_multiply
[rocm-libraries] ROCm/rocm-libraries#6302 (commit 8d419e8)
2026-04-10 15:18:02 +00:00
66_complex_contraction_bilinear
[rocm-libraries] ROCm/rocm-libraries#6323 (commit a668483)
2026-04-11 10:01:30 +00:00
67_gemm_microscaling
[CI, CK examples] Disable time_kernel for CI tests and examples (#3464)
2026-01-07 16:30:57 +01:00
68_gemm_add
[CK][Examples] Fixing stride issues in ck examples 14/65/68/69 by workaround - Bypassing hostTensor validation
2026-01-15 16:43:02 +01:00
69_gemm_add_relu
[CK][Examples] Fixing stride issues in ck examples 14/65/68/69 by workaround - Bypassing hostTensor validation
2026-01-15 16:43:02 +01:00
ck_tile
sparse_attn: R25 Step 1 A1 — per-warp PV-skip (paper Algorithm 1) + V0 instantiation
2026-05-18 06:13:38 -04:00
CMakeLists.txt
Build CK on Windows (#3458)
2026-01-14 07:31:45 -08:00
README.md
Add basic documentation structure (#1715)
2024-12-04 00:46:47 +01:00

README.md

Back to the main page

Composable Kernel examples

Reference in New Issue View Git Blame Copy Permalink