Skip to content

Commit 95525be

Browse files
committed
use CUDA archs 10.0, 10.0a, 10.0f for CUOBJDUMP_DEVICE_CODE_ONLY
1 parent f565c11 commit 95525be

File tree

1 file changed

+12
-6
lines changed

1 file changed

+12
-6
lines changed

test/framework/systemtools.py

Lines changed: 12 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -428,16 +428,22 @@
428428
CUOBJDUMP_DEVICE_CODE_ONLY = """
429429
Fatbin elf code:
430430
================
431-
arch = sm_90
432-
code version = [1,7]
431+
arch = sm_100f
432+
code version = [1,8]
433433
host = linux
434434
compile_size = 64bit
435-
compressed
436435
437436
Fatbin 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]
441447
host = linux
442448
compile_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 "

0 commit comments

Comments
 (0)