From 494896587fc013757715c23b11854aefdeedfbcf Mon Sep 17 00:00:00 2001 From: Fabi Eitel Date: Tue, 26 Apr 2022 14:57:54 +0200 Subject: [PATCH] Added Google Colab nvcc_args --- networks/channelnorm_package/setup.py | 6 +++++- networks/correlation_package/setup.py | 5 ++++- networks/resample2d_package/setup.py | 5 ++++- 3 files changed, 13 insertions(+), 3 deletions(-) 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(