File tree Expand file tree Collapse file tree 1 file changed +12
-6
lines changed
Expand file tree Collapse file tree 1 file changed +12
-6
lines changed Original file line number Diff line number Diff line change 428428CUOBJDUMP_DEVICE_CODE_ONLY = """
429429Fatbin elf code:
430430================
431- arch = sm_90
432- code version = [1,7 ]
431+ arch = sm_100f
432+ code version = [1,8 ]
433433host = linux
434434compile_size = 64bit
435- compressed
436435
437436Fatbin elf code:
438437================
439- arch = sm_90a
440- code version = [1,7]
438+ arch = sm_100
439+ code version = [1,8]
440+ host = linux
441+ compile_size = 64bit
442+
443+ Fatbin elf code:
444+ ================
445+ arch = sm_100a
446+ code version = [1,8]
441447host = linux
442448compile_size = 64bit"""
443449
@@ -1425,7 +1431,7 @@ def test_get_cuda_architectures(self):
14251431 self .assertIsNone (res_ptx )
14261432 fail_msg = "Pattern '%s' should be found in: %s" % (warning_regex_ptx .pattern , logtxt )
14271433 self .assertTrue (warning_regex_ptx .search (logtxt ), fail_msg )
1428- self .assertEqual (res_elf , ['9 .0' , '9 .0a' ])
1434+ self .assertEqual (res_elf , ['10 .0' , '10 .0a' , '10.0f ' ])
14291435
14301436 # Test case 6: call on CUDA shared lib which lacks an arch = sm_XX entry (should never happen)
14311437 warning_regex_elf = re .compile (r"WARNING Found Fatbin elf code section\(s\) in cuobjdump output for "
You can’t perform that action at this time.
0 commit comments