Add support for p100 in transformer kernels (#470)

add compute cap of 6.0, support p100

Add support for p100 in transformer kernels (#470)
add compute cap of 6.0, support p100
7ddfda85 · Jeff Rasley · GitHub · 1afca8f7 · 7ddfda85
隐藏空白更改
内联并排

Showing with 4 addition and 0 deletion

setup.py setup.py +4 -0

未找到文件。
--- 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__',