4 people 874a6b5a06 [ROCm] Refactor tuned gemms (#55001) ...

Signed-off-by: Andy Friedrich <afriedri@amd.com>
Signed-off-by: afriedri <afriedri@amd.com>
Co-authored-by: Douglas Lehr <91553416+dllehr-amd@users.noreply.github.com>
Co-authored-by: Shanshan Shen <467638484@qq.com>
Co-authored-by: mergify[bot] <37929162+mergify[bot]@users.noreply.github.com>

2026-09-21 16:30:16 +00:00

..

nvfp4_utils.py

Add pydocstyle to the ruff rules (#52136)

2026-09-16 16:44:29 +00:00

test_aiter_hipb_mm_linear_kernel.py

[CI] Ignore ruff D209, rejoin the docstrings it split, and silence incompatible-rule warnings (#57212)

2026-09-16 10:57:24 -07:00

test_allspark_gemm.py

[Quant][Kernel] Remove GPTQ Group/Dynamic Activation Ordering (#54809)

2026-09-08 12:45:12 -07:00

test_awq_triton.py

[Kernel][Test] Extend lightning_attn and awq_triton kernel tests to XPU (#43307)

2026-06-04 14:25:59 -04:00

test_awq.py

[Refactor] Remove dead quantization code and tests (#45454)

2026-06-17 16:12:01 -04:00

test_block_fp8.py

C3x SM100 FP8 blockwise - Pad activation scales to multiple of 4 (#57377)

2026-09-18 08:39:27 +00:00

test_cpu_fp8_scaled_mm.py

[CPU] Add CPU FP8 W8A8 linear/MoE support (#49942)

2026-09-18 17:59:42 +08:00

test_cutlass_scaled_mm.py

Add pydocstyle to the ruff rules (#52136)

2026-09-16 16:44:29 +00:00

test_cutlass_w4a8_moe.py

Add pydocstyle to the ruff rules (#52136)

2026-09-16 16:44:29 +00:00

test_cutlass_w4a8.py

Add pydocstyle to the ruff rules (#52136)

2026-09-16 16:44:29 +00:00

test_flashinfer_mxfp8_trtllm.py

[Kernel] Add FlashInfer TRTLLM MXFP8 linear backend (#52204)

2026-08-20 10:07:22 -04:00

test_flashinfer_nvfp4_scaled_mm.py

[Kernel] add Flashinfer cutedsl w4a16 linear (#53014)

2026-09-01 02:49:14 +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

Add pydocstyle to the ruff rules (#52136)

2026-09-16 16:44:29 +00:00

test_fp8_quant_group.py

[Bugfix] Initialize DeepGemmQuantScaleFMT oracle lazily; bound QuantFP8 UE8M0 packed path to group_size 128 (#51359)

2026-08-12 14:27:50 -04:00

test_fp8_quant.py

[CI] Bump mypy version (#34950)

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

test_gptq.py

[Quant][Kernel] Remove GPTQ Group/Dynamic Activation Ordering (#54809)

2026-09-08 12:45:12 -07: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_int4_emulation_moe.py

[feature]Add int4 quantization support for emulation moe backend (#48451)

2026-07-16 02:00:12 +00:00

test_int8_kernel.py

[CI] Ignore ruff D209, rejoin the docstrings it split, and silence incompatible-rule warnings (#57212)

2026-09-16 10:57:24 -07: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

[Core][Kernel] Share persistent workspaces for Marlin and Humming (#57421)

2026-09-20 17:28:33 +08:00

test_marlin_tile_padding.py

[Core][Kernel] Share persistent workspaces for Marlin and Humming (#57421)

2026-09-20 17:28:33 +08:00

test_mxfp4_kernel_selection.py

[Feature][Humming] Humming feature integration (#56685)

2026-09-20 08:11:01 -07:00

test_mxfp4_qutlass.py

[Platform] Deprecate seed_everything (#31659)

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

test_mxfp4_triton_ep.py

[CI] Ignore ruff D209, rejoin the docstrings it split, and silence incompatible-rule warnings (#57212)

2026-09-16 10:57:24 -07:00

test_mxfp6_kernel_selection.py

[CI] Ignore ruff D209, rejoin the docstrings it split, and silence incompatible-rule warnings (#57212)

2026-09-16 10:57:24 -07:00

test_nvfp4_emulation.py

Add pydocstyle to the ruff rules (#52136)

2026-09-16 16:44:29 +00:00

test_nvfp4_kernel_selection.py

[Perf][Quant][NVFP4] Prefer W4A4 linear kernels over weight-only ones on SM120/121 (#55170)

2026-09-08 15:06:45 +04:00

test_nvfp4_quant.py

Add pydocstyle to the ruff rules (#52136)

2026-09-16 16:44:29 +00: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

[CI] Ignore ruff D209, rejoin the docstrings it split, and silence incompatible-rule warnings (#57212)

2026-09-16 10:57:24 -07:00

test_quant_op_schema.py

[ROCm][CI] Move ROCm AITER quantization tests (#40938)

2026-08-18 14:50:35 +08:00

test_quantized_embedding.py

[CI] Ignore ruff D209, rejoin the docstrings it split, and silence incompatible-rule warnings (#57212)

2026-09-16 10:57:24 -07:00

test_rdna3_compile_guards.py

[Refactor][ROCm] Migrate the RDNA3 W4A16 MoE to the oracle/experts pa… (#55522)

2026-09-12 21:20:26 +08:00

test_rdna3_moe_w4a16.py

[Quant][Kernel] Remove GPTQ Group/Dynamic Activation Ordering (#54809)

2026-09-08 12:45:12 -07:00

test_rdna3_w4a16_selection.py

[Quant][Kernel] Remove GPTQ Group/Dynamic Activation Ordering (#54809)

2026-09-08 12:45:12 -07:00

test_rdna3_w4a16.py

[Quant][Kernel] Remove GPTQ Group/Dynamic Activation Ordering (#54809)

2026-09-08 12:45:12 -07:00

test_rdna_hybrid_w4a16.py

[ROCm][Perf] W4A16: keep skinny GEMM zero-points packed 4-bit (#54965)

2026-09-14 17:40:25 +08:00

test_rocm_aiter_grouped_quant.py

[CI] Ignore ruff D209, rejoin the docstrings it split, and silence incompatible-rule warnings (#57212)

2026-09-16 10:57:24 -07:00

test_rocm_compressed_tensors_w4a16.py

[CI] Retire Weight Loading smoke tests (#57398)

2026-09-17 16:04:03 -04:00

test_rocm_fp8.py

[ROCm][CI] Add More AITER quantization/MoE kernel tests (#49375)

2026-08-05 21:27:33 +00:00

test_rocm_mxfp4.py

[ROCm] Refactor tuned gemms (#55001)

2026-09-21 16:30:16 +00:00

test_rocm_mxfp8_linear.py

[Model] Support DeepSeek-V4.1-Flash (#56214)

2026-09-11 02:11:19 -07:00

test_rocm_skinny_gemms.py

[ROCm][Bugfix][Perf] Tune multi-stream shared experts use; wvSplitKrc fixes (#56098)

2026-09-10 15:14:22 -07:00

test_scaled_mm_kernel_selection.py

Add pydocstyle to the ruff rules (#52136)

2026-09-16 16:44:29 +00:00

test_silu_mul_nvfp4_quant.py

[Bugfix] Fix NVFP4 fused SiLU+mul scale allocation and global scale direction (#55643)

2026-09-08 08:26:07 -07:00

test_triton_scaled_mm.py

Add pydocstyle to the ruff rules (#52136)

2026-09-16 16:44:29 +00:00

test_triton_w4a16.py

[Quant][Kernel] Remove GPTQ Group/Dynamic Activation Ordering (#54809)

2026-09-08 12:45:12 -07:00

test_w4a16_kernel_selection.py

[Quant][Kernel] Remove GPTQ Group/Dynamic Activation Ordering (#54809)

2026-09-08 12:45:12 -07:00