You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
Is your feature request related to a problem? Please describe.
For multi-gpu training / inference, many some components only need to execute on specified ranks, so it would be nice to support the config for target rank ID.