diff --git a/networks/channelnorm_package/setup.py b/networks/channelnorm_package/setup.py index 5b9e86a4..bd290650 100755 --- a/networks/channelnorm_package/setup.py +++ b/networks/channelnorm_package/setup.py @@ -8,11 +8,15 @@ cxx_args = ['-std=c++11'] nvcc_args = [ + '-gencode', 'arch=compute_35,code=sm_35', + '-gencode', 'arch=compute_37,code=sm_37', + '-gencode', 'arch=compute_50,code=sm_50', '-gencode', 'arch=compute_52,code=sm_52', '-gencode', 'arch=compute_60,code=sm_60', '-gencode', 'arch=compute_61,code=sm_61', '-gencode', 'arch=compute_70,code=sm_70', - '-gencode', 'arch=compute_70,code=compute_70' + '-gencode', 'arch=compute_70,code=compute_70', + '-gencode', 'arch=compute_75,code=sm_75' ] setup( diff --git a/networks/correlation_package/setup.py b/networks/correlation_package/setup.py index 48b7d73a..556fb5df 100755 --- a/networks/correlation_package/setup.py +++ b/networks/correlation_package/setup.py @@ -8,12 +8,15 @@ cxx_args = ['-std=c++11'] nvcc_args = [ + '-gencode', 'arch=compute_35,code=sm_35', + '-gencode', 'arch=compute_37,code=sm_37', '-gencode', 'arch=compute_50,code=sm_50', '-gencode', 'arch=compute_52,code=sm_52', '-gencode', 'arch=compute_60,code=sm_60', '-gencode', 'arch=compute_61,code=sm_61', '-gencode', 'arch=compute_70,code=sm_70', - '-gencode', 'arch=compute_70,code=compute_70' + '-gencode', 'arch=compute_70,code=compute_70', + '-gencode', 'arch=compute_75,code=sm_75' ] setup( diff --git a/networks/resample2d_package/setup.py b/networks/resample2d_package/setup.py index bbedb255..fcfeaec4 100755 --- a/networks/resample2d_package/setup.py +++ b/networks/resample2d_package/setup.py @@ -8,12 +8,15 @@ cxx_args = ['-std=c++11'] nvcc_args = [ + '-gencode', 'arch=compute_35,code=sm_35', + '-gencode', 'arch=compute_37,code=sm_37', '-gencode', 'arch=compute_50,code=sm_50', '-gencode', 'arch=compute_52,code=sm_52', '-gencode', 'arch=compute_60,code=sm_60', '-gencode', 'arch=compute_61,code=sm_61', '-gencode', 'arch=compute_70,code=sm_70', - '-gencode', 'arch=compute_70,code=compute_70' + '-gencode', 'arch=compute_70,code=compute_70', + '-gencode', 'arch=compute_75,code=sm_75' ] setup(