Update (base update)

[ghstack-poisoned]
This commit is contained in:
Michael Lazos
2025-10-28 15:14:24 -07:00
parent 6dc807e0ac
commit d82cc9dcf9

View File

@ -5,6 +5,7 @@ from torch.testing._internal.common_utils import run_tests, skipIfTorchDynamo, T
class TestStream(TestCase):
@skipIfTorchDynamo()
def test_stream_create(self):
stream = torch.Stream(device="openreg")
self.assertEqual(stream.device_index, torch.openreg.current_device())