DeepSpeed is a deep learning optimization library that makes distributed training and inference easy, efficient, and effective.

0 0 56 Python

[build] fix computer capability arch flags, add PTX, handle PTX (#591)

* fix arch flags, add PTX

* bug fix

Co-authored-by: Jeff Rasley <jerasley@microsoft.com>

Stas Bekman committed 5y ago

8a184b6b1ddfe4bf41467c06b693817af80aa530

Parent: 0518252

Committed by GitHub <noreply@github.com> on 12/11/2020, 6:15:33 PM