Skip to content

GitLab

Unverified Commit 7ddfda85 authored by Jeff Rasley's avatar Jeff Rasley Committed by GitHub
Browse files

Add support for p100 in transformer kernels (#470) 

add compute cap of 6.0, support p100
parent 1afca8f7
Hide whitespace changes
Inline Side-by-side
Showing with 4 additions and 0 deletions
setup.py
View file @ 7ddfda85
......@@ -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__',
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment