Skip to content

GitLab

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

add compute cap of 6.0 to transformer kernels 

add compute cap of 6.0 to transformer kernels
parent e25f2a23
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 1 deletion
setup.py
View file @ b8eb40eb
......@@ -181,7 +181,7 @@ if BUILD_MASK & DS_BUILD_CPU_ADAM:
'-O3',
'--use_fast_math',
'-gencode',
'arch=compute_61,code=compute_61',
5 'arch=compute_61,code=compute_61',
'-gencode',
'arch=compute_70,code=compute_70',
'-std=c++14',
......@@ -216,6 +216,7 @@ if BUILD_MASK & DS_BUILD_TRANSFORMER:
'--use_fast_math',
'-gencode',
'arch=compute_61,code=compute_61',
'arch=compute_60,code=compute_60',
'-gencode',
'arch=compute_70,code=compute_70',
'-std=c++14',
......@@ -247,6 +248,7 @@ if BUILD_MASK & DS_BUILD_TRANSFORMER:
'--use_fast_math',
'-gencode',
'arch=compute_61,code=compute_61',
'arch=compute_60,code=compute_60',
'-gencode',
'arch=compute_70,code=compute_70',
'-std=c++14',
......
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