plasma-umass
diff --git a/‎test/torchtest.py
+2-1 b/‎test/torchtest.py
+2-1
diff --git a/‎tests/test_coverup_18.py
-43 b/‎tests/test_coverup_18.py
-43
diff --git a/‎tests/test_coverup_27.py
-33 b/‎tests/test_coverup_27.py
-33
diff --git a/‎tests/test_coverup_57.py
-64 b/‎tests/test_coverup_57.py
-64
@@ -4,8 +4,9 @@
 def torchtest():
     dtype = torch.float
     #device = torch.device("cpu")
-    device = torch.device("cuda:0")  # Uncomment this to run on GPU
+    # device = torch.device("cuda:0")  # Uncomment this to run on GPU
     # device = torch.device("cuda")  # Uncomment this to run on GPU
+    device = torch.device("mps")
 
     # Create Tensors to hold input and outputs.
     # By default, requires_grad=False, which indicates that we do not need to