| .. |
|
attention
|
[Fix] Replace deprecated T.gemm(..., wg_wait=...) with T.wgmma_gemm() + T.wait_wgmma()
|
2026-05-12 15:56:57 +08:00 |
|
elementwise
|
[Feat][Kernels] int/bool kernel coverage for MaskedFillScalar (#1411)
|
2026-05-12 09:57:31 +08:00 |
|
moe
|
[Feat][MoE] Add modular MoE interface: ABCs + FusedMoe orchestration (#1087)
|
2026-04-28 20:51:01 +08:00 |
|
norm
|
[Fix][NORM] enforce required weight/bias on GroupNorm/InstanceNorm forward (#1402)
|
2026-05-12 09:57:31 +08:00 |
|
reduction
|
[Fix][Reduction] reject bool from _validate_dim
|
2026-05-12 20:10:23 +08:00 |
|
__init__.py
|
[Feat][Attention] support GQA prefill fused RoPE (#1234)
|
2026-05-06 20:30:34 +08:00 |
|
convolution.py
|
[Refactor][CONV1D] Add dilation support to forward kernel (#1038)
|
2026-04-24 19:24:55 +08:00 |
|
da_cumsum.py
|
[BugFix][Mamba] Fix da_cumsum kernel to support dt_bias, softplus, and clamp (#1118)
|
2026-05-09 18:00:38 +08:00 |
|
deltanet.py
|
[Fix][Ops] Complete GqaSlidingWindow rename and enable RUF022 (#967)
|
2026-04-14 21:45:16 +08:00 |
|
deltanet_recurrence.py
|
[Refactor] Rename base modules to <module>_base.py convention (#950)
|
2026-04-13 18:13:32 +08:00 |
|
dropout.py
|
[Refactor] Rename base modules to <module>_base.py convention (#950)
|
2026-04-13 18:13:32 +08:00 |
|
engram.py
|
[Fix][Ops] Complete GqaSlidingWindow rename and enable RUF022 (#967)
|
2026-04-14 21:45:16 +08:00 |
|
engram_decode.py
|
[Refactor] Rename base modules to <module>_base.py convention (#950)
|
2026-04-13 18:13:32 +08:00 |
|
fft.py
|
[Refactor] Rename base modules to <module>_base.py convention (#950)
|
2026-04-13 18:13:32 +08:00 |
|
fp8_lighting_indexer.py
|
[Refactor] Rename base modules to <module>_base.py convention (#950)
|
2026-04-13 18:13:32 +08:00 |
|
fp8_quant.py
|
[Refactor] Rename base modules to <module>_base.py convention (#950)
|
2026-04-13 18:13:32 +08:00 |
|
gated_deltanet.py
|
[Fix][Ops] Complete GqaSlidingWindow rename and enable RUF022 (#967)
|
2026-04-14 21:45:16 +08:00 |
|
gated_linear_attn.py
|
[Refactor][Ops] Consolidate op files into one file per op family (#960)
|
2026-04-14 17:23:52 +08:00 |
|
gemm.py
|
[Refactor] Rename base modules to <module>_base.py convention (#950)
|
2026-04-13 18:13:32 +08:00 |
|
gla.py
|
[Fix][Ops] Complete GqaSlidingWindow rename and enable RUF022 (#967)
|
2026-04-14 21:45:16 +08:00 |
|
grouped_gemm.py
|
[Refactor] Rename base modules to <module>_base.py convention (#950)
|
2026-04-13 18:13:32 +08:00 |
|
mhc.py
|
[Fix][Ops] Complete GqaSlidingWindow rename and enable RUF022 (#967)
|
2026-04-14 21:45:16 +08:00 |
|
op_base.py
|
[Chore][Release] merge testbed into main: op-family alignment + new ops (#1380)
|
2026-05-09 00:03:37 +08:00 |
|
pool.py
|
[Refactor] Rename base modules to <module>_base.py convention (#950)
|
2026-04-13 18:13:32 +08:00 |
|
rope.py
|
[Feat][Attention] support GQA prefill fused RoPE (#1234)
|
2026-05-06 20:30:34 +08:00 |
|
ssd_chunk_scan.py
|
[BugFix] Align Mamba2 kernels to official Mamba-2 triton interface + refactor for performance (#1015)
|
2026-04-28 10:20:37 +08:00 |
|
ssd_chunk_state.py
|
[Refactor] Rename base modules to <module>_base.py convention (#950)
|
2026-04-13 18:13:32 +08:00 |
|
ssd_decode.py
|
[BugFix] Align Mamba2 kernels to official Mamba-2 triton interface + refactor for performance (#1015)
|
2026-04-28 10:20:37 +08:00 |
|
ssd_state_passing.py
|
[Refactor] Rename base modules to <module>_base.py convention (#950)
|
2026-04-13 18:13:32 +08:00 |
|
topk_selector.py
|
[Refactor] Rename base modules to <module>_base.py convention (#950)
|
2026-04-13 18:13:32 +08:00 |
8aeb36bf4a