From 7ddfda85263890615791b63955b9f5ed28ea2eff Mon Sep 17 00:00:00 2001 From: Jeff Rasley Date: Wed, 14 Oct 2020 10:44:16 -0700 Subject: [PATCH] Add support for p100 in transformer kernels (#470) add compute cap of 6.0, support p100 --- setup.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/setup.py b/setup.py index 2363ee4d..1364caab 100755 --- a/setup.py +++ b/setup.py @@ -217,6 +217,8 @@ if BUILD_MASK & DS_BUILD_TRANSFORMER: '-gencode', 'arch=compute_61,code=compute_61', '-gencode', + 'arch=compute_60,code=compute_60', + '-gencode', 'arch=compute_70,code=compute_70', '-std=c++14', '-U__CUDA_NO_HALF_OPERATORS__', @@ -248,6 +250,8 @@ if BUILD_MASK & DS_BUILD_TRANSFORMER: '-gencode', 'arch=compute_61,code=compute_61', '-gencode', + 'arch=compute_60,code=compute_60', + '-gencode', 'arch=compute_70,code=compute_70', '-std=c++14', '-U__CUDA_NO_HALF_OPERATORS__', -- GitLab