From 3ca846e3dfbbf50ae70034083206e72de860cb62 Mon Sep 17 00:00:00 2001 From: fredwang Date: Mon, 9 Aug 2021 08:56:43 +0800 Subject: [PATCH] Update setup.py for the nvcc compilation options for half float. Remove compute_35 and compute_50 because 1. they do not support half float; 2. google colab's GPU has been updated from K80 (compute_35) to T4 (compute_75). --- setup.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/setup.py b/setup.py index f7a3f19281..a4dc17bc19 100644 --- a/setup.py +++ b/setup.py @@ -267,9 +267,9 @@ def prepare_extension_options(): extra_compile_args = {'gcc': get_cpp_flags()} if with_cuda: - cuda9_gencode = (' -gencode arch=compute_35,code=sm_35' - ' -gencode arch=compute_50,code=sm_50' - ' -gencode arch=compute_60,code=sm_60' + # compute_35 and compute_50 are removed because 1. they do not support half float; + # 2. google colab's GPU has been updated from K80 (compute_35) to T4 (compute_75). + cuda9_gencode = (' -gencode arch=compute_60,code=sm_60' ' -gencode arch=compute_70,code=sm_70') cuda10_gencode = ' -gencode arch=compute_75,code=sm_75' cuda11_gencode = ' -gencode arch=compute_80,code=sm_80'