3 people 06d020bb6e [Bugfix] Fix SM121 (DGX Spark) exclusion from Marlin/CUTLASS FP8 paths (#35568) ...

Signed-off-by: Blake Ledden <blake@secondnaturecomputing.com>
Co-authored-by: Claude Opus 4.6 <noreply@anthropic.com>
Co-authored-by: Pavani Majety <pmajety@nvidia.com>

2026-05-15 10:59:00 -07:00

..

attention

[Perf] Optimize MLA attention _v_up_proj bmm by removing additional copy (#42561)

2026-05-15 08:14:26 -07:00

fla

[Bugfix] Fix GDN KKT precision loss on Hopper GPUs by aligning tl.dot operand layout with WGMMA (#42076)

2026-05-09 13:08:46 +00:00

fused_moe

[ROCm] Widen OAI Triton MoE capability range to include gfx12 (RDNA4) (#37826)

2026-05-15 07:59:57 -07:00

mamba

[Hybrid] Warmup Mamba2 SSD kernel (#39822)

2026-05-12 12:46:22 +00:00

pooler

[Perf][2/n] Eliminate GPU<->CPU syncs in pooling code (#41433)

2026-05-05 02:43:25 +00:00

quantization

[Bugfix] Fix SM121 (DGX Spark) exclusion from Marlin/CUTLASS FP8 paths (#35568)

2026-05-15 10:59:00 -07:00

rotary_embedding

[CPU] Fix rotary embedding for CPU without flash-attn ops (#42225)

2026-05-12 15:05:35 +00:00

__init__.py

…

activation.py

[Models][Gemma3/Gemma4] Support hidden_act variants in gated MLP (#40588)

2026-05-08 11:29:11 -07:00

attention_layer_base.py

[1/N][Attention] Restructure attention: move files (#31916)

2026-01-09 13:10:24 -08:00

batch_invariant.py

[Refactor] Cleanup batch invariant dead code (#41993)

2026-05-11 10:48:39 -04:00

conv.py

[Misc] Always use forward_mulmat for Conv3d on newer versions of torch. (#38487)

2026-03-30 05:39:41 +00:00

deepseek_compressor.py

[New Model][ROCm] Add AMD support for DeepSeek V4 (#40871)

2026-05-05 08:55:37 -07:00

deepseek_v4_attention.py

[ROCm][DSv4] implement flash sparse mla with triton kernels (#41812)

2026-05-11 09:27:11 -07:00

kda.py

[Misc] Replace mamba_type string literals with MambaAttentionBackendEnum (#41188)

2026-05-11 03:59:36 +00:00

layernorm.py

[Aiter][ROCm] RMSNormGated+GroupedQuantFP8 fusion (#40710)

2026-05-14 15:37:09 -04:00

lightning_attn.py

[Bugfix] Use null block (0) for padded block table entries (#35431)

2026-03-30 14:02:51 -07:00

linear.py

[Quant] Consolidate GPTQ: rename gptq_marlin.py to auto_gptq.py (#38288)

2026-05-15 08:25:52 +08:00

logits_processor.py

[PluggableLayer][3/N] Apply PluggableLayer to llm_head and vocab embedding layer (#33465)

2026-04-10 16:12:59 +08:00

mhc.py

DeepSeekV4-Pro enable cuda graph full and piecewise mode (#42604)

2026-05-15 01:45:30 -07:00

mla.py

[Feature]: IndexCache support for DSA models (#37735)

2026-04-29 15:15:35 -04:00

resampler.py

Update Optional[x] -> x | None and Union[x, y] to x | y (#26633)

2025-10-12 09:51:31 -07:00

sparse_attn_indexer.py

[New Model][ROCm] Add AMD support for DeepSeek V4 (#40871)

2026-05-05 08:55:37 -07:00

utils.py

[DeepSeek] Use torch.mm for bf16xbf16->fp32 gemm (#41300)

2026-04-30 16:28:57 -07:00

vocab_parallel_embedding.py

Bug/test eagle dp v0 (#38938)

2026-04-13 20:50:08 +00:00