Skip to content

Commit f677e95

Browse files
author
akutalev
committed
YNMT: Correct CUDA arcs for Hopper & Blackwell
commit_hash:579ca37e57672c2da4c91d8ef78eefcd2ada9c92
1 parent 5cc4e55 commit f677e95

File tree

1 file changed

+6
-0
lines changed

1 file changed

+6
-0
lines changed

build/ymake_conf.py

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -2540,6 +2540,12 @@ def auto_cuda_architectures(self):
25402540
if version >= (11, 8):
25412541
architectures.extend(['sm_89', 'sm_90'])
25422542

2543+
if version >= (12, 0):
2544+
architectures.append('sm_90a')
2545+
2546+
if version >= (12, 8):
2547+
architectures.extend(['sm_100', 'sm_100a', 'sm_120', 'sm_120a'])
2548+
25432549
return ':'.join(architectures)
25442550

25452551
def auto_use_arcadia_cuda(self):

0 commit comments

Comments
 (0)