DeepSpeed
be4ffb82 - Reduction Kernel Utility (#2436)

Commit
2 years ago
Reduction Kernel Utility (#2436) * Initial reduction_utils.h implementation * Add initialization helper, ensures correct min/max behavior * Remove unnecessary warp sync
Author
Parents
Loading