Skip to content

Commit e2ad6c1

Browse files
committed
fix test failure due to using init_cuda.arch instead of sm_{init_cuda.arch}
1 parent 4575865 commit e2ad6c1

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

cuda_core/tests/test_module.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -381,7 +381,7 @@ def test_object_code_load_object_from_file(get_saxpy_object, tmp_path):
381381

382382

383383
def test_object_code_load_object_with_linker(get_saxpy_object, init_cuda):
384-
arch = init_cuda.arch
384+
arch = f"sm_{init_cuda.arch}"
385385
kernel_ptx = Program(
386386
r"""
387387
extern __device__ float saxpy_step(float a, float x, float y);

0 commit comments

Comments
 (0)