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
Treat meta as CPU device, so we can run the meta on CPU with MetaSparseDispatchMode (#1617)
Summary:
Pull Request resolved: #1617
When we use meta, it's not treated as CPU tensor. But the use case is that we want to process the weights on CPU first.
So we set use_cpu as true when device is true in `IntNBitTableBatchedEmbeddingBagsCodegen`
Reviewed By: jianyuh, nanoax
Differential Revision: D43633336
fbshipit-source-id: bfb5f52c17f6ecf8bedb4f47a85aabcf35b87416
0 commit comments