From 6cece06c55cab280749265bff0c8e8e199c86cfe Mon Sep 17 00:00:00 2001 From: vmoens Date: Wed, 19 Jun 2024 09:15:29 -0700 Subject: [PATCH] amend --- test/test_specs.py | 1 + 1 file changed, 1 insertion(+) diff --git a/test/test_specs.py b/test/test_specs.py index 83a688c7ed5..6b779811f1d 100644 --- 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")