Try to identify device

lab-cosmo · Aug 31, 2024 · 6da2942 · 6da2942
1 parent 39f45bf
commit 6da2942
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/sphericart-torch/python/tests/test_autograd.py b/sphericart-torch/python/tests/test_autograd.py
@@ -122,7 +122,9 @@ def test_second_derivative_error(xyz):
     # case 1: autograd.grad raises an error
     # being the first time the second derivatives are requested, and since
     # `backward_second_derivatives=False`, a warning is also displayed
-    print("WAT ")
+    print(
+        "WAT ", sph.device, torch.device("mps").type, torch.backends.mps.is_available()
+    )
     try:
         torch.autograd.grad(
             outputs=l1,