Stas Bekman 8a184b6b1d [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>		2020-12-11 10:15:33 -08:00
..
__init__.py	Add compute capability 8.0 if on cuda 11+ (#572)	2020-12-02 17:22:16 -08:00
builder.py	[build] fix computer capability arch flags, add PTX, handle PTX (#591)	2020-12-11 10:15:33 -08:00
cpu_adam.py	Dependency pruning (#528)	2020-11-13 17:36:33 -08:00
fused_adam.py	[build] make builder smarter and configurable wrt compute capabilities + docs (#578)	2020-12-07 12:08:41 -08:00
fused_lamb.py	[build] make builder smarter and configurable wrt compute capabilities + docs (#578)	2020-12-07 12:08:41 -08:00
sparse_attn.py	DeepSpeed JIT op + PyPI support (#496)	2020-11-12 11:51:38 -08:00
stochastic_transformer.py	DeepSpeed JIT op + PyPI support (#496)	2020-11-12 11:51:38 -08:00
transformer.py	DeepSpeed JIT op + PyPI support (#496)	2020-11-12 11:51:38 -08:00
utils.py	DeepSpeed JIT op + PyPI support (#496)	2020-11-12 11:51:38 -08:00