Skip to content

[bugfix] fix correctness of kjt.lengths when ShardedEmbeddingBag’s pooling_type is mean and shard_type is row_wise#106

Merged
tiankongdeguiji merged 4 commits intoalibaba:masterfrom
tiankongdeguiji:debug/row_wise_mean
Feb 13, 2025
Merged

[bugfix] fix correctness of kjt.lengths when ShardedEmbeddingBag’s pooling_type is mean and shard_type is row_wise#106
tiankongdeguiji merged 4 commits intoalibaba:masterfrom
tiankongdeguiji:debug/row_wise_mean

Conversation

@tiankongdeguiji
Copy link
Collaborator

No description provided.

@tiankongdeguiji tiankongdeguiji changed the title [bugfix] fix correctness of kjt.lengths when ShardedEmbeddingBag pooling_type is mean and shard_type is row_wise [bugfix] fix correctness of kjt.lengths when ShardedEmbeddingBag’s pooling_type is mean and shard_type is row_wise Feb 13, 2025
@tiankongdeguiji tiankongdeguiji merged commit 76016ec into alibaba:master Feb 13, 2025
5 checks passed
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.

3 participants