sm_35 isn't supported in CUDA 12

tinygrad · Nov 26, 2023 · d2c5c7b · d2c5c7b
1 parent 14af655
commit d2c5c7b
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/test/test_cuda.py b/test/test_cuda.py
@@ -26,10 +26,10 @@ def test_has_methods(self):
 
   def test_compile_fail(self):
     with self.assertRaises(RuntimeError):
-      cuda_compile("__device__ void test() { {", ["--gpu-architecture=sm_35"], CUDACompile, check)
+      cuda_compile("__device__ void test() { {", ["--gpu-architecture=sm_60"], CUDACompile, check)
 
   def test_compile(self):
-    prg = cuda_compile("__device__ int test() { return 42; }", ["--gpu-architecture=sm_35"], CUDACompile, check)
+    prg = cuda_compile("__device__ int test() { return 42; }", ["--gpu-architecture=sm_60"], CUDACompile, check)
     assert len(prg) > 10
 
 @unittest.skipIf(CI, "cuda doesn't work in CI")