@@ -158,7 +158,7 @@ def test_object_code_load_ptx_from_file(get_saxpy_kernel_ptx, tmp_path):
158158 ptx_file .write_bytes (ptx )
159159 mod_obj = ObjectCode .from_ptx (str (ptx_file ), symbol_mapping = sym_map )
160160 assert mod_obj .code == str (ptx_file )
161- assert mod_obj ._code_type == "ptx"
161+ assert mod_obj .code_type == "ptx"
162162 if not Program ._can_load_generated_ptx ():
163163 pytest .skip ("PTX version too new for current driver" )
164164 mod_obj .get_kernel ("saxpy<double>" ) # force loading
@@ -198,7 +198,7 @@ def test_object_code_load_ltoir(get_saxpy_kernel_ltoir):
198198 assert isinstance (ltoir , bytes )
199199 mod_obj = ObjectCode .from_ltoir (ltoir , symbol_mapping = sym_map )
200200 assert mod_obj .code == ltoir
201- assert mod_obj ._code_type == "ltoir"
201+ assert mod_obj .code_type == "ltoir"
202202 # ltoir doesn't support kernel retrieval directly as it's used for linking
203203 assert mod_obj ._handle is None
204204 # Test that get_kernel fails for unsupported code type
@@ -215,7 +215,7 @@ def test_object_code_load_ltoir_from_file(get_saxpy_kernel_ltoir, tmp_path):
215215 ltoir_file .write_bytes (ltoir )
216216 mod_obj = ObjectCode .from_ltoir (str (ltoir_file ), symbol_mapping = sym_map )
217217 assert mod_obj .code == str (ltoir_file )
218- assert mod_obj ._code_type == "ltoir"
218+ assert mod_obj .code_type == "ltoir"
219219 # ltoir doesn't support kernel retrieval directly as it's used for linking
220220 assert mod_obj ._handle is None
221221
@@ -418,4 +418,4 @@ def test_module_serialization_roundtrip(get_saxpy_kernel_cubin):
418418 assert isinstance (result , ObjectCode )
419419 assert objcode .code == result .code
420420 assert objcode ._sym_map == result ._sym_map
421- assert objcode ._code_type == result ._code_type
421+ assert objcode .code_type == result .code_type
0 commit comments