Skip to content

Commit c224dfc

Browse files
committed
avoid linking kernel_ptx with cubin that triggers nvrtc nvjitlink version mismatch in ci blackwell gpu
1 parent e2ad6c1 commit c224dfc

1 file changed

Lines changed: 3 additions & 3 deletions

File tree

cuda_core/tests/test_module.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -382,7 +382,7 @@ def test_object_code_load_object_from_file(get_saxpy_object, tmp_path):
382382

383383
def test_object_code_load_object_with_linker(get_saxpy_object, init_cuda):
384384
arch = f"sm_{init_cuda.arch}"
385-
kernel_ptx = Program(
385+
kernel_code = Program(
386386
r"""
387387
extern __device__ float saxpy_step(float a, float x, float y);
388388
extern "C" __global__ void linked_kernel(float a, float x, float y, float* out) {
@@ -391,9 +391,9 @@ def test_object_code_load_object_with_linker(get_saxpy_object, init_cuda):
391391
""",
392392
"c++",
393393
ProgramOptions(relocatable_device_code=True, arch=arch),
394-
).compile("ptx")
394+
).compile("cubin")
395395
linked = Linker(
396-
kernel_ptx,
396+
kernel_code,
397397
ObjectCode.from_object(get_saxpy_object),
398398
options=LinkerOptions(arch=arch),
399399
).link("cubin")

0 commit comments

Comments
 (0)