@@ -172,25 +172,29 @@ def append_nvcc_threads(nvcc_extra_args):
172172 "Note: make sure nvcc has a supported version by running nvcc -V."
173173 )
174174
175+ if bare_metal_version <= Version ("12.9" ):
176+ cc_flag .append ("-gencode" )
177+ cc_flag .append ("arch=compute_53,code=sm_53" )
178+ cc_flag .append ("-gencode" )
179+ cc_flag .append ("arch=compute_62,code=sm_62" )
180+ cc_flag .append ("-gencode" )
181+ cc_flag .append ("arch=compute_70,code=sm_70" )
182+ cc_flag .append ("-gencode" )
183+ cc_flag .append ("arch=compute_72,code=sm_72" )
175184 cc_flag .append ("-gencode" )
176- cc_flag .append ("arch=compute_53,code=sm_53" )
177- cc_flag .append ("-gencode" )
178- cc_flag .append ("arch=compute_62,code=sm_62" )
179- cc_flag .append ("-gencode" )
180- cc_flag .append ("arch=compute_70,code=sm_70" )
181- cc_flag .append ("-gencode" )
182- cc_flag .append ("arch=compute_72,code=sm_72" )
185+ cc_flag .append ("arch=compute_75,code=sm_75" )
183186 cc_flag .append ("-gencode" )
184187 cc_flag .append ("arch=compute_80,code=sm_80" )
185188 cc_flag .append ("-gencode" )
186189 cc_flag .append ("arch=compute_87,code=sm_87" )
187-
188190 if bare_metal_version >= Version ("11.8" ):
189191 cc_flag .append ("-gencode" )
190192 cc_flag .append ("arch=compute_90,code=sm_90" )
191193 if bare_metal_version >= Version ("12.8" ):
192194 cc_flag .append ("-gencode" )
193195 cc_flag .append ("arch=compute_100,code=sm_100" )
196+ cc_flag .append ("-gencode" )
197+ cc_flag .append ("arch=compute_120,code=sm_120" )
194198
195199
196200 # HACK: The compiler flag -D_GLIBCXX_USE_CXX11_ABI is set to be the same as
0 commit comments