Skip to content

D
DeepSpeed

Greenplum / DeepSpeed
上一次同步 大约 1 年

通知 10
Star 0
Fork 0
D
DeepSpeed

体验新版 GitCode，发现更多精彩内容 >>

未验证 提交 b8eb40eb 编写于 作者: J Jeff Rasley 提交者： GitHub
浏览文件
操作

add compute cap of 6.0 to transformer kernels 

add compute cap of 6.0 to transformer kernels
上级 e25f2a23
隐藏空白更改
内联 并排
Showing with 3 addition and 1 deletion
setup.py
浏览文件 @ 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% 或 .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑！
想要评论请 注册 或