Thomas Parnell 2f925e5777

[Kernel] Unified Triton kernel that doesn't distinguish between prefill + decode (#16828) ...

Signed-off-by: Thomas Parnell <tpa@zurich.ibm.com>
Signed-off-by: Lucas Wilkinson <lwilkinson@neuralmagic.com>
Co-authored-by: Lucas Wilkinson <lwilkinson@neuralmagic.com>

2025-05-06 18:21:48 -04:00

..

attention

[Bugfix] Fix triton import with local TritonPlaceholder (#17446)

2025-05-06 17:53:09 +08:00

core

Categorize tests/kernels/ based on kernel type (#16799)

2025-04-23 09:21:07 -04:00

mamba

Categorize tests/kernels/ based on kernel type (#16799)

2025-04-23 09:21:07 -04:00

moe

[Kernel] some optimizations for dense marlin and moe marlin (#16850)

2025-05-05 09:39:30 -07:00

quantization

[Misc] Fix ScalarType float4 naming (#17690)

2025-05-06 01:07:15 -07:00

__init__.py

[CI/Build] Move test_utils.py to tests/utils.py (#4425)

2024-05-13 23:50:09 +09:00

allclose_default.py

[Misc] Add SPDX-License-Identifier headers to python source files (#12628)

2025-02-02 11:58:18 -08:00

quant_utils.py

Add missing rocm_skinny_gemms kernel test to CI (#17060)

2025-04-24 07:49:37 -07:00

test_cutlass_mla_decode.py

[NVIDIA] Support Cutlass MLA for Blackwell GPUs (#16032)

2025-04-27 06:29:21 -07:00

test_fused_quant_activation.py

[torch.compile] Add torch inductor pass for fusing silu_and_mul with subsequent scaled_fp8_quant operations (#10867)

2025-05-01 07:59:28 -07:00

test_triton_flash_attention.py

[Kernel][Triton][FP8] Adding fp8 and variable length sequence support to Triton FAv2 kernel (#12591)

2025-04-27 00:35:08 +00:00

test_triton_unified_attention.py

[Kernel] Unified Triton kernel that doesn't distinguish between prefill + decode (#16828)

2025-05-06 18:21:48 -04:00

utils.py

[Misc] Replace os environ to monkeypatch in test suite (#14516)

2025-03-16 20:35:57 -07:00