split cpu parts in permute_pooled_embedding_ops for cpu_only #987
Conversation
|
Hi @RabbitWhite1! Thank you for your pull request and welcome to our community. Action RequiredIn order to merge any pull request (code, docs, etc.), we require contributors to sign our Contributor License Agreement, and we don't seem to have one on file for you. ProcessIn order for us to review and merge your suggested changes, please sign at https://code.facebook.com/cla. If you are contributing on behalf of someone else (eg your employer), the individual CLA may not be sufficient and your employer may need to sign the corporate CLA. Once the CLA is signed, our tooling will perform checks and validations. Afterwards, the pull request will be tagged with If you have received this in error or have any questions, please contact us at cla@fb.com. Thanks! |
|
Thank you for signing our Contributor License Agreement. We can now accept your code for this (and any) Meta Open Source project. Thanks! |
|
@jianyuh has imported this pull request. If you are a Meta employee, you can view this diff on Phabricator. |
|
cc @geyyer : please take a look at this PR. |
|
Fixed duplicated m.def in permute_pooled_embedding_ops_gpu.cpp |
|
@RabbitWhite1, thanks for contributing and noticing the possible issue with cpu-only build! To avoid compatibility issue we have to create an empty file permute_pooled_embs_function.h and then move the class template PermutePooledEmbsFunction there. The PR looks good, could you move permute_pooled_embedding_ops_utils.h contents to permute_pooled_embs_function.h and fix dependencies? |
Great! I've moved and committed. |
|
@jianyuh has imported this pull request. If you are a Meta employee, you can view this diff on Phabricator. |
|
@geyyer has imported this pull request. If you are a Meta employee, you can view this diff on Phabricator. |
…te1/FBGEMM into cpu_permute_pooled_emb
|
@geyyer has imported this pull request. If you are a Meta employee, you can view this diff on Phabricator. |
Summary: Catching up with pytorch#987 Differential Revision: D35261607 fbshipit-source-id: c806f035bfcb4f8872d083431dc7819f6532b7d8
|
This pull request has been reverted by 386a473. |
As specified in CMakeLists.txt, "src/permute_pooled_embedding_ops_gpu.cpp" will only compile when "NOT FBGEMM_CPU_ONLY", which means the method "permute_pooled_embs_auto_grad" won't be generated when --cpu_only. However, this method is used by torchrec's column_wise sharding.
The pr mentioned in #950 cannot work because of not using m.def to define permute_pooled_embs_auto_grad.