Commit 875403e
refactor: update dpsk fused_moe test [2] (#2097)
<!-- .github/pull_request_template.md -->
## π Description
Add shuffling and blockmajorK layout in dpskv3 fused_moe fp8_blockscaled
tests.
## π Related Issues
<!-- Link any related issues here -->
## π Pull Request Checklist
Thank you for contributing to FlashInfer! Before we review your pull
request, please make sure the following items are complete.
### β
Pre-commit Checks
- [x] I have installed `pre-commit` by running `pip install pre-commit`
(or used your preferred method).
- [x] I have installed the hooks with `pre-commit install`.
- [x] I have run the hooks manually with `pre-commit run --all-files`
and fixed any reported issues.
> If you are unsure about how to set up `pre-commit`, see [the
pre-commit documentation](https://pre-commit.com/).
## π§ͺ Tests
- [x] Tests have been added or updated as needed.
- [x] All tests are passing (`unittest`, etc.).
## Reviewer Notes
<!-- Optional: anything you'd like reviewers to focus on, concerns, etc.
-->
<!-- This is an auto-generated comment: release notes by coderabbit.ai
-->
## Summary by CodeRabbit
* **Tests**
* Expanded MoE test suite with per-expert weight shuffling, optional
block-layout conversion, selectable weight-processing modes, and dynamic
kernel flags.
* Added a reference FP8 block-scale validation path and centralized
accuracy checks for clearer correctness verification.
* **Refactor**
* Centralized test utilities: quantization mode and test-skip logic
moved into shared helpers for consistent gating across MoE tests.
<!-- end of auto-generated comment: release notes by coderabbit.ai -->
---------
Co-authored-by: Zihao Ye <[email protected]>1 parent a9f71bd commit 875403e
File tree
3 files changed
+392
-176
lines changed- tests/moe
3 files changed
+392
-176
lines changed
0 commit comments