Skip to content

Commit 7320786

Browse files
committed
ensure kernel has C linkage
1 parent 7fcb5ad commit 7320786

File tree

1 file changed

+2
-6
lines changed

1 file changed

+2
-6
lines changed

cuda_core/tests/test_module.py

Lines changed: 2 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -15,13 +15,9 @@
1515

1616
@pytest.mark.xfail(not can_load_generated_ptx(), reason="PTX version too new")
1717
def test_get_kernel():
18-
kernel = """
19-
extern __device__ int B();
20-
extern __device__ int C(int a, int b);
21-
__global__ void A() { int result = C(B(), 1);}
22-
"""
18+
kernel = """extern "C" __global__ void ABC() { }"""
2319
object_code = Program(kernel, "c++").compile("ptx", options=("-rdc=true",))
2420
assert object_code._handle is None
25-
kernel = object_code.get_kernel("A")
21+
kernel = object_code.get_kernel("ABC")
2622
assert object_code._handle is not None
2723
assert kernel._handle is not None

0 commit comments

Comments
 (0)