remove cuda tensors

Author: mark14wu

tests/test_autotune_add.py

@@ -50,8 +50,8 @@ def test_autotune_add_inrange():
     This test uses n_elements = 128, matching the size of the input tensors.
     It should NOT cause any out-of-bound access.
     """
-    x = torch.randn(128, device="cuda")
-    y = torch.randn(128, device="cuda")
+    x = torch.randn(128)
+    y = torch.randn(128)
     out = torch.empty_like(x)
 
     # The kernel launch uses n_elements=128, aligned with the tensor size.
@@ -66,8 +66,8 @@ def test_autotune_add_out_of_bound():
     This test deliberately sets n_elements = 256, exceeding the actual buffer size (128).
     It will likely cause out-of-bound reads/writes, which may trigger errors or warnings.
     """
-    x = torch.randn(128, device="cuda")
-    y = torch.randn(128, device="cuda")
+    x = torch.randn(128)
+    y = torch.randn(128)
     out = torch.empty_like(x)
 
     # The kernel launch uses n_elements=256, exceeding the valid tensor size.