amend

pytorch · Jun 19, 2024 · 6cece06 · 6cece06
1 parent 180abff
commit 6cece06
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/test/test_specs.py b/test/test_specs.py
@@ -3729,6 +3729,7 @@ def test_device_ordinal():
         moneh=moneh,
         mdisc=mdisc,
         shape=(-1, 1, 2),
+        device=device,
     )
     assert spec.device == torch.device("cuda:0")