liangel-02 6b5c389ee3 expose flex block size for batch invariant mode (#41252) ...

Signed-off-by: Angel Li <liangel@meta.com>

2026-05-13 14:11:57 -07:00

..

attention

expose flex block size for batch invariant mode (#41252)

2026-05-13 14:11:57 -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

[Quantization] Rework quantization_config to use QuantKey and allow for activation override (#41566)

2026-05-13 16:58:32 -04: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

[Quantization] Rework quantization_config to use QuantKey and allow for activation override (#41566)

2026-05-13 16:58:32 -04: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

[vLLM IR] 2/N fused_add_rms_norm and maybe_inplace overload (#36823)

2026-05-01 23:41:15 -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

[Quantization] Add ModelOpt NVFP4 W4A16 (4-bit weights, fp16/bf16 activations) support (#41769)

2026-05-09 21:15:50 +00: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

[Bugfix] [ROCm] [DSV4] [Perf] Add aiter mhc support (#41946)

2026-05-13 21:43:15 +08: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