add cubin for Turing
@@ -636,7 +636,13 @@ def _nvcc_gencode_options(cuda_version):
arch_list = ['compute_30',
'compute_50']
- if cuda_version >= 9000:
+ if cuda_version >= 10000:
+ arch_list += [('compute_60', 'sm_60'),
+ ('compute_61', 'sm_61'),
+ ('compute_70', 'sm_70'),
+ ('compute_75', 'sm_75'),
+ 'compute_70']
+ elif cuda_version >= 9000:
arch_list += [('compute_60', 'sm_60'),
('compute_61', 'sm_61'),
('compute_70', 'sm_70'),
