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 年之前