Skip to content

[CI] Remove legacy dequantize gemm test#2013

Merged
LeiWang1999 merged 1 commit into
tile-ai:mainfrom
LeiWang1999:remove_test_0405
Apr 5, 2026
Merged

[CI] Remove legacy dequantize gemm test#2013
LeiWang1999 merged 1 commit into
tile-ai:mainfrom
LeiWang1999:remove_test_0405

Conversation

@LeiWang1999
Copy link
Copy Markdown
Member

@LeiWang1999 LeiWang1999 commented Apr 5, 2026

as title.

Summary by CodeRabbit

  • Chores
    • Removed example script demonstrating grouped matrix multiplication with quantization support
    • Removed associated test and regression test cases for the deleted example
    • Removed an alternative GEMM example regression execution path

…regression and test files after its deletion. This cleanup ensures that the codebase remains consistent and free of unused imports.
@github-actions
Copy link
Copy Markdown

github-actions Bot commented Apr 5, 2026

👋 Hi! Thank you for contributing to the TileLang project.

Please remember to run pre-commit run --all-files in the root directory of the project to ensure your changes are properly linted and formatted. This will help ensure your contribution passes the format check.

We appreciate you taking this step! Our team will review your contribution, and we look forward to your awesome work! 🚀

@coderabbitai
Copy link
Copy Markdown
Contributor

coderabbitai Bot commented Apr 5, 2026

No actionable comments were generated in the recent review. 🎉

ℹ️ Recent review info
⚙️ Run configuration

Configuration used: defaults

Review profile: CHILL

Plan: Pro

Run ID: 0549db59-6e3e-4395-b90a-fb5cf4a798a1

📥 Commits

Reviewing files that changed from the base of the PR and between 01c714d and 204fd2d.

📒 Files selected for processing (4)
  • examples/dequantize_gemm/example_dequant_groupedgemm_bf16_mxfp4_hopper.py
  • examples/dequantize_gemm/regression_example_dequantize_gemm.py
  • examples/dequantize_gemm/test_example_dequantize_gemm.py
  • examples/gemm/regression_example_gemm.py
💤 Files with no reviewable changes (4)
  • examples/dequantize_gemm/regression_example_dequantize_gemm.py
  • examples/dequantize_gemm/test_example_dequantize_gemm.py
  • examples/gemm/regression_example_gemm.py
  • examples/dequantize_gemm/example_dequant_groupedgemm_bf16_mxfp4_hopper.py

📝 Walkthrough

Walkthrough

This PR removes an example GEMM kernel script with FP4 dequantization for Hopper GPUs, along with its associated test function and regression wrapper across three test/regression files.

Changes

Cohort / File(s) Summary
Example Dequantization Script Removal
examples/dequantize_gemm/example_dequant_groupedgemm_bf16_mxfp4_hopper.py
Deleted entire example file (~571 lines) including: autotuning config generator, JIT-compiled grouped GEMM kernel with FP4 dequantization and bias fusion paths, dequantization helper macros, reference implementation, benchmarking logic, and CLI entrypoint.
Test & Regression Wrapper Cleanup
examples/dequantize_gemm/regression_example_dequantize_gemm.py, examples/dequantize_gemm/test_example_dequantize_gemm.py
Removed imports and corresponding wrapper functions: regression_example_dequant_groupedgemm_bf16_mxfp4_hopper() and test_example_dequant_groupedgemm_bf16_mxfp4_hopper().
GEMM Regression Cleanup
examples/gemm/regression_example_gemm.py
Removed regression_example_gemm_schedule() wrapper function and its import, eliminating schedule-based GEMM regression path.

Estimated code review effort

🎯 2 (Simple) | ⏱️ ~8 minutes

Possibly related PRs

Suggested reviewers

  • tzj-fxz

Poem

🐰 The example hops away into the void,
No more MoE-like GEMMs to deploy,
Regression paths trimmed with care,
Hopper's fancy kernels fade in the air,
Cleanup done, the code is lighter! 🌙

🚥 Pre-merge checks | ✅ 3
✅ Passed checks (3 passed)
Check name Status Explanation
Description Check ✅ Passed Check skipped - CodeRabbit’s high-level summary is enabled.
Title check ✅ Passed The title clearly summarizes the main change: removal of a legacy dequantize GEMM test, which aligns with all modified files deleting references to the example_dequant_groupedgemm_bf16_mxfp4_hopper example and its associated test infrastructure.
Docstring Coverage ✅ Passed No functions found in the changed files to evaluate docstring coverage. Skipping docstring coverage check.

✏️ Tip: You can configure your own custom pre-merge checks in the settings.

✨ Finishing Touches
🧪 Generate unit tests (beta)
  • Create PR with unit tests

Thanks for using CodeRabbit! It's free for OSS, and your support helps us grow. If you like it, consider giving us a shout-out.

❤️ Share

Comment @coderabbitai help to get the list of available commands and usage tips.

@LeiWang1999 LeiWang1999 merged commit bb79425 into tile-ai:main Apr 5, 2026
6 of 7 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant