From 41c4407c8ce716cea0b7ff0a4979ee142942cca5 Mon Sep 17 00:00:00 2001 From: ksimpson Date: Mon, 6 Jan 2025 15:28:56 -0800 Subject: [PATCH] leverage fixture in fixture --- cuda_core/tests/test_module.py | 9 +-------- 1 file changed, 1 insertion(+), 8 deletions(-) diff --git a/cuda_core/tests/test_module.py b/cuda_core/tests/test_module.py index f6418eaa..7931d2c3 100644 --- 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", "saxpy"), )