microsoft
/
DeepSpeed
зеркало из https://github.com/microsoft/DeepSpeed.git
1
0
Форкнуть 0
Код Задачи Пакеты Проекты Релизы Вики Активность

Add support for p100 in transformer kernels (#470) 

add compute cap of 6.0, support p100
This commit is contained in:
Jeff Rasley 2020-10-14 10:44:16 -07:00 коммит произвёл GitHub
Родитель 1afca8f722
Коммит 7ddfda8526
Не найден ключ, соответствующий данной подписи
Идентификатор ключа GPG: 4AEE18F83AFDEB23
1 изменённых файлов: 4 добавлений и 0 удалений
Показать статистику Скачать Patch файл Скачать Diff файл Показать все файлы Свернуть все файлы

4
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__',