leverage fixture in fixture

NVIDIA · Jan 6, 2025 · 41c4407 · 41c4407
1 parent d4e966e
commit 41c4407
Showing 1 changed file with 1 addition and 8 deletions.
diff --git a/cuda_core/tests/test_module.py b/cuda_core/tests/test_module.py
@@ -29,18 +29,11 @@ def get_saxpy_kernel(init_cuda):
     }
     """
 
-    dev = Device()
-    dev.set_current()
-    dev.create_stream()
-
     # prepare program
     prog = Program(code, code_type="c++")
     mod = prog.compile(
         "cubin",
-        options=(
-            "-std=c++11",
-            "-arch=sm_" + "".join(f"{i}" for i in dev.compute_capability),
-        ),
+        options=("-arch=sm_" + "".join(f"{i}" for i in Device().compute_capability),),
         name_expressions=("saxpy<float>", "saxpy<double>"),
     )