Skip to content

ManagedCollisionEmbeddingBagCollectionSharder cannot support CPU #2958

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
YuandongXu-hotstar opened this issue May 8, 2025 · 2 comments
Open

Comments

@YuandongXu-hotstar
Copy link

YuandongXu-hotstar commented May 8, 2025

Does ManagedCollisionEmbeddingBagCollectionSharder only support cuda?
I want to use ManagedCollisionEmbeddingBagCollection in CPU only environment.
But I cannot resolve the "RuntimeError: No available sharding type and compute kernel combination after applying user provided constraints for XXX"
What should I do?

for shard types that each sharder supports in CPU device type:
EmbeddingBagCollectionSharder: ['data_parallel', 'table_wise', 'column_wise', 'table_column_wise']
FeatureProcessedEmbeddingBagCollectionSharder: ['data_parallel', 'table_wise', 'column_wise', 'table_column_wise']
EmbeddingCollectionSharder: ['data_parallel', 'table_wise', 'column_wise', 'row_wise']
FusedEmbeddingBagCollectionSharder: ['data_parallel', 'table_wise', 'column_wise', 'table_column_wise']
QuantEmbeddingBagCollectionSharder: ['table_wise', 'row_wise', 'column_wise']
QuantEmbeddingCollectionSharder: ['table_wise', 'row_wise', 'column_wise']
ManagedCollisionEmbeddingBagCollectionSharder: [] ----------------------> this is empty list
ManagedCollisionEmbeddingCollectionSharder: ['row_wise']
QuantManagedCollisionEmbeddingCollectionSharder: ['row_wise']

Can I use EmbeddingBagCollectionSharder when I use ManagedCollisionEmbeddingBagCollection for the embeddings? What would be the behavior?

@iamzainhuda
Copy link
Contributor

@kausv could you take a look at this?

@kausv
Copy link
Contributor

kausv commented Jun 6, 2025

Ack starting the work on this

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

No branches or pull requests

3 participants