no cuda tests if no cuda available

Author: inikishev

pyproject.toml

@@ -1,3 +1,10 @@
+# NEW VERSION TUTORIAL
+# STEP 1 - CHANGE VERSION IN THIS FILE
+# STEP 2 - CREATE TAG WITH THAT VERSIOn
+# STEP 3 - PUSH TO GITHUB
+# STEP 4 - PUSH TAG TO GITHUB
+# STEP 5 - WAIT AND SEE THAT TESTS PASS ON GITHUB AND THAT IT DIDNT FIND ANOTHER REASON TO SKIP PIP
+
 [build-system]
 requires = ["setuptools >= 61.0"]
 build-backend = "setuptools.build_meta"

tests/test_identical.py

@@ -89,6 +89,7 @@ def _assert_identical_merge_closure(opt_fn: Callable, device, steps: int):
 
 def _assert_identical_device(opt_fn: Callable, merge: bool, use_closure: bool, steps: int):
     """checks that trajectories match on cpu and cuda."""
+    if not torch.cuda.is_available(): return
     cpu, cpu_opt = _get_trajectory(opt_fn, _BOOTH_X0.clone().cpu(), merge=merge, use_closure=use_closure, steps=steps)
     cuda, cuda_opt = _get_trajectory(opt_fn, _BOOTH_X0.clone().cuda(), merge=merge, use_closure=use_closure, steps=steps)
     _compare_trajectories(cpu_opt, cpu, cuda_opt, cuda.to(cpu))