Add support for p100 in transformer kernels (#470)

add compute cap of 6.0, support p100
microsoft · Oct 14, 2020 · 7ddfda8 · 7ddfda8
1 parent 1afca8f
commit 7ddfda8
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/setup.py b/setup.py
@@ -217,6 +217,8 @@ def available_vector_instructions():
                               '-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 @@ def available_vector_instructions():
                               '-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__',