KimmiShi 98c96e790b Update flops profiler to handle attn and __matmul__ (#4724) | 8 months ago | |
---|---|---|
.. | ||
flops_profiler | 98c96e790b Update flops profiler to handle attn and __matmul__ (#4724) | 8 months ago |
__init__.py | b361c72761 Update DeepSpeed copyright license to Apache 2.0 (#3111) | 1 year ago |
config.py | 460bec4679 flops_profiler: add option recompute_fwd_factor for the case of activation recompute (#3362) | 1 year ago |
constants.py | 460bec4679 flops_profiler: add option recompute_fwd_factor for the case of activation recompute (#3362) | 1 year ago |