DeepSpeed

[build] fix computer capability arch flags, add PTX, handle PTX

#591

Merged