We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 8413877 commit c6566e0Copy full SHA for c6566e0
1 file changed
tests/unit_tests/moe/test_backend_config.py
@@ -143,7 +143,8 @@ def test_enable_deepep_false_sets_dispatcher_torch(self):
143
warnings.simplefilter("always")
144
config = BackendConfig(enable_deepep=False)
145
assert config.dispatcher == "torch"
146
- assert config.experts == "torch_mm" # experts unchanged when enable_deepep=False
+ expected_experts = "torch_mm" if torch.cuda.is_available() else "torch"
147
+ assert config.experts == expected_experts # experts unchanged when enable_deepep=False
148
assert config.enable_deepep is None # Should be cleared after conversion
149
assert len(w) == 1
150
assert issubclass(w[0].category, DeprecationWarning)
0 commit comments