@@ -43,13 +43,13 @@ def _is_nvvm_available():
4343precheck_nvvm_ir = """target triple = "nvptx64-unknown-cuda"
4444target datalayout = "e-p:64:64:64-i1:8:8-i8:8:8-i16:16:16-i32:32:32-i64:64:64-i128:128:128-f32:32:32-f64:64:64-v16:16:16-v32:32:32-v64:64:64-v128:128:128-n16:32:64"
4545
46- define void @dummy_kernel() {
46+ define void @dummy_kernel() {{
4747 entry:
4848 ret void
49- }
49+ }}
5050
5151!nvvm.annotations = !{{!0}}
52- !0 = !{{void (i32* )* @simple , !"kernel", i32 1}}
52+ !0 = !{{void ()* @dummy_kernel , !"kernel", i32 1}}
5353
5454!nvvmir.version = !{{!1}}
5555!1 = !{{i32 {major}, i32 {minor}, i32 {debug_major}, i32 {debug_minor}}}
@@ -87,6 +87,7 @@ def _get_libnvvm_version_for_tests():
8787 program = nvvm .create_program ()
8888 try :
8989 major , minor , debug_major , debug_minor = nvvm .ir_version ()
90+ global precheck_nvvm_ir
9091 precheck_nvvm_ir = precheck_nvvm_ir .format (major = major , minor = minor , debug_major = debug_major , debug_minor = debug_minor )
9192 precheck_ir_bytes = precheck_nvvm_ir .encode ("utf-8" )
9293 nvvm .add_module_to_program (program , precheck_ir_bytes , len (precheck_ir_bytes ), "precheck.ll" )
@@ -103,7 +104,6 @@ def _get_libnvvm_version_for_tests():
103104 cuda_version = get_minimal_required_cuda_ver_from_ptx_ver (ptx_version )
104105 _libnvvm_version = cuda_version
105106 return _libnvvm_version
106-
107107 finally :
108108 nvvm .destroy_program (program )
109109
0 commit comments