[CI Bug] Fix ci issue #40637, Kernels FusedMoE Layer Test (2 H100s): test_moe_layer.py::test_moe_layer#40639
[CI Bug] Fix ci issue #40637, Kernels FusedMoE Layer Test (2 H100s): test_moe_layer.py::test_moe_layer#40639yewentao256 wants to merge 1 commit intomainfrom
Conversation
yewentao256
requested review from
WoosukKwon,
mgoin and
tlrmchlsmth
as code owners
![claude[bot]](https://avatars.githubusercontent.com/in/1236702?s=60&v=4){kind=small}
There was a problem hiding this comment.
Claude Code Review
This repository is configured for manual code reviews. Comment @claude review to trigger a review and subscribe this PR to future pushes, or @claude review once for a one-time review.
Tip: disable this comment in your organization's Code Review settings.
![gemini-code-assist[bot]](https://avatars.githubusercontent.com/in/956858?s=60&v=4){kind=small}
There was a problem hiding this comment.
Code Review
This pull request modifies the MoE layer tests to conditionally clear the DeepEP all2all manager's cache based on device capability. The reviewer noted that the current implementation only targets Blackwell (major version 10) devices, whereas the PR's intent includes H100 (major version 9) support. A suggestion was provided to broaden the capability check and use more idiomatic membership testing for the backend.
|
For reference: #35077 (comment) |
|
Close as fixed by #35077 (comment) |
2 participants
Purpose
Fix ci issue #40637, seems issue introduced from #35077
Test
Covered in CI