зеркало из https://github.com/microsoft/DeepSpeed.git
revert previous (accidental) change
This commit is contained in:
Родитель
b8eb40eb67
Коммит
1afca8f722
4
setup.py
4
setup.py
|
@ -181,7 +181,7 @@ if BUILD_MASK & DS_BUILD_CPU_ADAM:
|
|||
'-O3',
|
||||
'--use_fast_math',
|
||||
'-gencode',
|
||||
5 'arch=compute_61,code=compute_61',
|
||||
'arch=compute_61,code=compute_61',
|
||||
'-gencode',
|
||||
'arch=compute_70,code=compute_70',
|
||||
'-std=c++14',
|
||||
|
@ -216,7 +216,6 @@ 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',
|
||||
|
@ -248,7 +247,6 @@ 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',
|
||||
|
|
Загрузка…
Ссылка в новой задаче