Correct Volta+Tegra -> Xavier.
This commit is contained in:
parent
e54fd996bb
commit
6d15594d44
|
@ -158,7 +158,7 @@ class CudaModule(ExtensionModule):
|
||||||
cuda_limit_gpu_architecture = '7.0' # noqa: E221
|
cuda_limit_gpu_architecture = '7.0' # noqa: E221
|
||||||
|
|
||||||
if version_compare(cuda_version, '>=9.0'):
|
if version_compare(cuda_version, '>=9.0'):
|
||||||
cuda_known_gpu_architectures += ['Volta', 'Volta+Tegra'] # noqa: E221
|
cuda_known_gpu_architectures += ['Volta', 'Xavier'] # noqa: E221
|
||||||
cuda_common_gpu_architectures += ['7.0', '7.0+PTX'] # noqa: E221
|
cuda_common_gpu_architectures += ['7.0', '7.0+PTX'] # noqa: E221
|
||||||
cuda_all_gpu_architectures += ['7.0', '7.0+PTX', '7.2', '7.2+PTX'] # noqa: E221
|
cuda_all_gpu_architectures += ['7.0', '7.0+PTX', '7.2', '7.2+PTX'] # noqa: E221
|
||||||
|
|
||||||
|
@ -225,7 +225,7 @@ class CudaModule(ExtensionModule):
|
||||||
'Pascal': (['6.0', '6.1'], ['6.1']),
|
'Pascal': (['6.0', '6.1'], ['6.1']),
|
||||||
'Pascal+Tegra': (['6.2'], []),
|
'Pascal+Tegra': (['6.2'], []),
|
||||||
'Volta': (['7.0'], ['7.0']),
|
'Volta': (['7.0'], ['7.0']),
|
||||||
'Volta+Tegra': (['7.2'], []),
|
'Xavier': (['7.2'], []),
|
||||||
'Turing': (['7.5'], ['7.5']),
|
'Turing': (['7.5'], ['7.5']),
|
||||||
}.get(arch_name, (None, None))
|
}.get(arch_name, (None, None))
|
||||||
|
|
||||||
|
|
Loading…
Reference in New Issue