We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 95d509b commit 103cd0fCopy full SHA for 103cd0f
test/test_operations.py
@@ -92,6 +92,7 @@ def onlyOnCPU(fn):
92
accelerator = os.environ.get("PJRT_DEVICE").lower()
93
return unittest.skipIf(accelerator != "cpu", "PJRT_DEVICE=CUDA required")(fn)
94
95
+
96
def onlyOnCUDA(fn):
97
98
return unittest.skipIf(accelerator != "cuda", "PJRT_DEVICE=CUDA required")(fn)
0 commit comments