PiPPy
PiPPy copied to clipboard
[SPMD][Fusion] add bucket size/ num_bytes policy for fusion
currently we run fusion based on an integer policy, where integer maps to total number of comm calls to fuse. need to add a bucket size policy handler to setup for fixed and variable optimization policies.