[Bug] cpu_only fbgemm won't compile "permute_pooled_embs_auto_grad"

[RabbitWhite1]

The used fbgemm @ c3a26e1 is buggy when compiling with "--cpu_only".

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 column_wise sharding.

To correctly support "--cpu_only", see the pr in FBGEMM#987

[jianyuh]

Thanks @RabbitWhite1 for reporting this issue! We are working on fixing this issue with pytorch/FBGEMM#950 as you mentioned. Currently the major blocker on landing the PR is due to some internal backward compatibility issues. Will push on landing it soon. cc @geyyer

[geyyer]

PR pytorch/FBGEMM#987 fixing the issue was landed.

[colin2328]

Thanks @geyyer . Closing this issue. @RabbitWhite1 feel free to file another issue if needed