Skip to content

Commit c6566e0

Browse files
committed
fix
Signed-off-by: Hemil Desai <hemild@nvidia.com>
1 parent 8413877 commit c6566e0

1 file changed

Lines changed: 2 additions & 1 deletion

File tree

tests/unit_tests/moe/test_backend_config.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -143,7 +143,8 @@ def test_enable_deepep_false_sets_dispatcher_torch(self):
143143
warnings.simplefilter("always")
144144
config = BackendConfig(enable_deepep=False)
145145
assert config.dispatcher == "torch"
146-
assert config.experts == "torch_mm" # experts unchanged when enable_deepep=False
146+
expected_experts = "torch_mm" if torch.cuda.is_available() else "torch"
147+
assert config.experts == expected_experts # experts unchanged when enable_deepep=False
147148
assert config.enable_deepep is None # Should be cleared after conversion
148149
assert len(w) == 1
149150
assert issubclass(w[0].category, DeprecationWarning)

0 commit comments

Comments
 (0)