Chen Zhang e20c92bb61 [Kernel] Move attn_type to Attention.__init__() (#11690) ...

Signed-off-by: Chen Zhang <zhangch99@outlook.com>

2025-01-07 00:11:28 +08:00

..

__init__.py

…

allclose_default.py

…

conftest.py

…

quant_utils.py

…

test_activation.py

…

test_aqlm.py

…

test_attention_selector.py

[platforms] enable platform plugins (#11602)

2024-12-30 20:24:45 +08:00

test_attention.py

…

test_awq_marlin.py

…

test_awq_triton.py

…

test_awq.py

…

test_block_fp8.py

[Model] [Quantization] Support deepseek_v3 w8a8 fp8 block-wise quantization (#11523)

2024-12-26 15:33:30 -08:00

test_blocksparse_attention.py

…

test_cache.py

…

test_cascade_flash_attn.py

[V1] Implement Cascade Attention (#11635)

2025-01-01 21:56:46 +09:00

test_causal_conv1d.py

[BugFix][Kernel]: fix illegal memory access in causal_conv1d when conv_states is None (#10928)

2024-12-08 01:21:18 +08:00

test_cutlass.py

…

test_encoder_decoder_attn.py

[Kernel] Move attn_type to Attention.__init__() (#11690)

2025-01-07 00:11:28 +08:00

test_flash_attn.py

[Kernel] Use out arg in flash_attn_varlen_func (#10811)

2024-12-01 17:55:39 -08:00

test_flashinfer.py

…

test_fp8_quant.py

…

test_fused_quant_layernorm.py

[torch.compile] Dynamic fp8 + rms_norm fusion (#10906)

2024-12-13 03:19:23 +00:00

test_ggml.py

…

test_gguf.py

…

test_gptq.py

…

test_int8_quant.py

…

test_layernorm.py

…

test_machete_mm.py

…

test_mamba_ssm.py

…

test_marlin_gemm.py

[Bugfix] Marlin 2:4 temp fix for large M dim (>256) (#10464)

2024-11-19 19:40:33 -08:00

test_moe.py

…

test_permute_cols.py

…

test_pos_encoding.py

…

test_prefix_prefill.py

[Bugfix] Fix chunked prefill with model dtype float32 on Turing Devices (#9850)

2024-11-25 12:23:32 -05:00

test_rotary_embedding.py

…

test_semi_structured.py

[Bugfix][Build/CI] Fix sparse CUTLASS compilation on CUDA [12.0, 12.2) (#11311)

2024-12-19 02:43:30 +00:00

test_triton_scaled_mm.py

…

test_utils.py

…

utils.py

[Kernel] Move attn_type to Attention.__init__() (#11690)

2025-01-07 00:11:28 +08:00