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 f8d269a commit e48ce4dCopy full SHA for e48ce4d
tests/conftest.py
@@ -149,7 +149,7 @@ def xp(
149
try:
150
device = jax.devices("cuda")[0]
151
except RuntimeError:
152
- pytest.skip("no cuda device available")
+ pytest.skip("no CUDA device available")
153
else:
154
device = jax.devices("cpu")[0]
155
jax.config.update("jax_default_device", device)
@@ -158,7 +158,7 @@ def xp(
158
import torch.cuda
159
160
if not torch.cuda.is_available():
161
162
xp.set_default_device("cuda")
163
164
elif library == Backend.TORCH: # CPU
0 commit comments