3 people 363fc84407 Integrate flashinfer b12x MoE and FP4 GEMM kernels for SM120/121 (#40082) ...

Signed-off-by: Meenakshi Venkataraman <meenakshiv@nvidia.com>
Co-authored-by: Claude Sonnet 4.6 <noreply@anthropic.com>

2026-05-20 17:21:11 +00:00

..

nvfp4_utils.py

Add nvfp4 support to reshape_and_cache_flash (#37332)

2026-04-17 07:28:00 -07:00

test_allspark_gemm.py

[Hardware] Replace torch.cuda.synchronize() api with torch.accelerator.synchronize (#36085)

2026-03-05 10:36:39 +00:00

test_awq_triton.py

[Platform] Deprecate seed_everything (#31659)

2026-01-04 18:34:04 -08:00

test_awq.py

[Bugfix] awq_gemm: fix argument order swap (#30364)

2025-12-14 18:15:37 +08:00

test_block_fp8.py

[W8A8 Block Linear Refactor][2/N] Remove W8A8Fp8BlockLinearOp and adopt Fp8 block linear kernel selections. (#33892)

2026-04-09 08:50:39 +08:00

test_block_int8.py

[Misc] Make SchedulerConfig.max_model_len init-only (#28733)

2025-11-15 01:59:31 -08:00

test_cpu_fp8_scaled_mm.py

[CPU] Add FP8 W8A16 linear support (#41186)

2026-05-06 07:05:27 +00:00

test_cutlass_scaled_mm.py

[CUDA][CUTLASS] Enable cutlass scaled mm for non-compatible sizes (#41868)

2026-05-08 15:58:05 -07:00

test_cutlass_w4a8_moe.py

[Hardware] Replace torch.cuda.synchronize() api with torch.accelerator.synchronize (#36085)

2026-03-05 10:36:39 +00:00

test_cutlass_w4a8.py

[CI/Build][AMD] Skip quantization kernels tests that require CUTLASS or e4m3fn when not supported by platform (#30020)

2025-12-10 02:28:37 +00:00

test_flashinfer_nvfp4_scaled_mm.py

Integrate flashinfer b12x MoE and FP4 GEMM kernels for SM120/121 (#40082)

2026-05-20 17:21:11 +00:00

test_flashinfer_scaled_mm.py

[Platform] Deprecate seed_everything (#31659)

2026-01-04 18:34:04 -08:00

test_fp8_min_max_helper.py

[Bugfix][Hardware][AMD] Consolidate FP8 min/max values helper function (#31106)

2026-01-07 06:55:03 +00:00

test_fp8_quant_group.py

[Misc] Fix Current vLLM config is not set. warnings, assert to avoid issues in the future (#31747)

2026-01-08 15:20:49 -08:00

test_fp8_quant.py

[CI] Bump mypy version (#34950)

2026-03-04 20:55:11 +00:00

test_ggml.py

Convert formatting to use ruff instead of yapf + isort (#26247)

2025-10-05 07:06:22 -07:00

test_gguf.py

[Platform] Deprecate seed_everything (#31659)

2026-01-04 18:34:04 -08:00

test_gptq.py

[Kernel] Add GPTQv2 format support for low-bit or asymmetric quantization, by adapting gptq_gemm (#26092)

2025-10-23 23:26:13 -04:00

test_hadacore.py

[CI/Build][AMD] Skip marlin, machete, and hadacore tests since these require _C functions not defined for ROCm (#30109)

2025-12-06 12:54:17 +08:00

test_int8_kernel.py

[Misc] Fix Current vLLM config is not set. warnings, assert to avoid issues in the future (#31747)

2026-01-08 15:20:49 -08:00

test_int8_quant.py

[Platform] Deprecate seed_everything (#31659)

2026-01-04 18:34:04 -08:00

test_machete_mm.py

[Hardware] Replace torch.cuda.device_count/current_device/set_device API (#36145)

2026-03-12 07:57:47 -07:00

test_marlin_gemm.py

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

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

test_mxfp4_qutlass.py

[Platform] Deprecate seed_everything (#31659)

2026-01-04 18:34:04 -08:00

test_mxfp4_triton_ep.py

[MoE] Move GPT OSS Triton kernel experts into fused_moe/experts/ (#39007)

2026-04-14 19:27:39 +00:00

test_nvfp4_emulation.py

[NVFP4][Hopper/AMD Instinct] Add Triton kernels for NVFP4 dequantization and QDQ emulation (#40033)

2026-04-30 17:35:48 -04:00

test_nvfp4_quant.py

[Perf] Padded nvfp4 quant kernel to remove additional copy, 2.4%~5.7% e2e performance improvement (#42774)

2026-05-18 16:38:12 -07:00

test_nvfp4_qutlass.py

[Platform] Deprecate seed_everything (#31659)

2026-01-04 18:34:04 -08:00

test_nvfp4_scaled_mm.py

[Platform] Deprecate seed_everything (#31659)

2026-01-04 18:34:04 -08:00

test_per_token_group_quant.py

Faster per-token fp8 group quant packed kernel for blackwell (#41326)

2026-04-30 18:09:55 -07:00

test_rocm_compressed_tensors_w4a16.py

[Kernel][Hardware][AMD] Add TritonW4A16LinearKernel for ROCm (#37352)

2026-04-10 10:25:27 +00:00

test_rocm_skinny_gemms.py

[ROCm] Enable wvSplitK skinny GEMM kernel for RDNA4/gfx1x decode (#34709)

2026-03-20 10:11:23 -05:00

test_scaled_mm_kernel_selection.py

[OOT] Add OOT support for linear kernel. (#37989)

2026-03-31 14:33:21 +08:00

test_silu_mul_nvfp4_quant.py

[Misc] Fix Current vLLM config is not set. warnings, assert to avoid issues in the future (#31747)

2026-01-08 15:20:49 -08:00

test_triton_scaled_mm.py

[Platform] Deprecate seed_everything (#31659)

2026-01-04 18:34:04 -08:00

test_triton_w4a16.py

[Kernel][Hardware][AMD] Add TritonW4A16LinearKernel for ROCm (#37352)

2026-04-10 10:25:27 +00:00

test_w4a16_kernel_selection.py

[Kernel][Hardware][AMD] Add TritonW4A16LinearKernel for ROCm (#37352)

2026-04-10 10:25:27 +00:00