Skip to content

Commit 0b2356c

Browse files
authored
add more arches (#78318)
1 parent 4128126 commit 0b2356c

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

python/paddle/utils/cpp_extension/extension_utils.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -469,6 +469,7 @@ def _get_cuda_arch_flags(cflags: list[str] | None = None) -> list[str]:
469469
'10.0a',
470470
'10.1',
471471
'10.1a',
472+
'10.3',
472473
'12.0',
473474
'12.0a',
474475
]

0 commit comments

Comments
 (0)