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