Logan Adams 170b46e8b1 Add conditional on torch version for scaled_dot_product_attention (#6517) | 1 月之前 | |
---|---|---|
.. | ||
flops_profiler | 170b46e8b1 Add conditional on torch version for scaled_dot_product_attention (#6517) | 1 月之前 |
__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 年之前 |