DeepSpeed/op_builder
Olatunji Ruwase 91e15593ea
Control ds_report output (#1622)
* Control ds_report output with two flags --hide_operators and --hide_errors_and_warnings
Separate cli and function entry points to ds_report

* Formatting fixes
2021-12-08 16:29:13 -08:00
..
__init__.py Add cpu adagrad (#1358) 2021-10-26 17:55:17 +00:00
async_io.py Control ds_report output (#1622) 2021-12-08 16:29:13 -08:00
builder.py Control ds_report output (#1622) 2021-12-08 16:29:13 -08:00
cpu_adagrad.py Control ds_report output (#1622) 2021-12-08 16:29:13 -08:00
cpu_adam.py Control ds_report output (#1622) 2021-12-08 16:29:13 -08:00
fused_adam.py Remove hard torch dependency at install (#1166) 2021-06-16 14:18:37 -07:00
fused_lamb.py Remove hard torch dependency at install (#1166) 2021-06-16 14:18:37 -07:00
quantizer.py Remove hard torch dependency at install (#1166) 2021-06-16 14:18:37 -07:00
sparse_attn.py Control ds_report output (#1622) 2021-12-08 16:29:13 -08:00
stochastic_transformer.py Remove hard torch dependency at install (#1166) 2021-06-16 14:18:37 -07:00
transformer.py Remove hard torch dependency at install (#1166) 2021-06-16 14:18:37 -07:00
transformer_inference.py Remove hard torch dependency at install (#1166) 2021-06-16 14:18:37 -07:00
utils.py ZeRO 3 Offload (#834) 2021-03-08 12:54:54 -08:00