Skip to content

Conversation

@esmeetu
Copy link
Member

@esmeetu esmeetu commented Mar 6, 2024

After #2221, tensor_model_parallel_all_gather seems not being used.
It's been replaced by tensor_model_parallel_gather. Correct me if i miss something. @zhuohan123

@zhuohan123
Copy link
Member

Let's keep this primitive for now? Some future models (e.g., some special implementation of MoE) may need this primitive.

@esmeetu esmeetu closed this Mar 7, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants