compile_kernel remove cp from graph test

2025-10-27 17:54:55 +08:00 · 2025-09-26 13:09:39 -07:00
1 changed files with 1 additions and 3 deletions
--- a/test/test_cuda.py
+++ b/test/test_cuda.py
@ -7289,9 +7289,7 @@ class TestCudaDeviceParametrized(TestCase):
        """
        from torch.cuda import _compile_kernel

-        spin_wait_kernel = _compile_kernel(
-            kernel_source, "wait_for_cpu", compute_capability="70"
-        )
+        spin_wait_kernel = _compile_kernel(kernel_source, "wait_for_cpu")

        x = torch.ones(4, device="cuda")
        x_cpu = torch.zeros(x.shape, device="cpu").pin_memory()