Fix TorchRec OSS stable version (pytorch#946)

Summary:
Pull Request resolved: pytorch#946

TorchRec OSS stable version still uses permute_sparse_data. Restore the old API for backward compatibility.

Reviewed By: yixin94

Differential Revision: D34383729

fbshipit-source-id: c08789028635bd0646ab2265091465f93210c1fd

fbgemm_gpu/src/sparse_ops_cpu.cpp

@@ -1617,6 +1617,8 @@ std::tuple<Tensor, Tensor> embedding_bag_rowwise_prune(
 } // namespace fbgemm_gpu
 
 TORCH_LIBRARY_FRAGMENT(fbgemm, m) {
+  m.def(
+      "permute_sparse_data(Tensor permute, Tensor lengths, Tensor values, Tensor? weights=None, int? permuted_lengths_sum=None) -> (Tensor, Tensor, Tensor?)");
   m.def(
       "permute_2D_sparse_data(Tensor permute, Tensor lengths, Tensor values, Tensor? weights=None, int? permuted_lengths_sum=None) -> (Tensor, Tensor, Tensor?)");
   m.def(
@@ -1658,6 +1660,8 @@ TORCH_LIBRARY_FRAGMENT(fbgemm, m) {
 }
 
 TORCH_LIBRARY_IMPL(fbgemm, CPU, m) {
+  DISPATCH_TO_CPU(
+      "permute_sparse_data", fbgemm_gpu::permute_2D_sparse_data_cpu);
   DISPATCH_TO_CPU(
       "permute_2D_sparse_data", fbgemm_gpu::permute_2D_sparse_data_cpu);
   DISPATCH_TO_CPU(

fbgemm_gpu/src/sparse_ops_gpu.cpp

@@ -164,6 +164,8 @@ std::vector<Tensor> stacked_jagged_2d_to_dense_gpu(
 } // namespace fbgemm_gpu
 
 TORCH_LIBRARY_IMPL(fbgemm, CUDA, m) {
+  DISPATCH_TO_CUDA(
+      "permute_sparse_data", fbgemm_gpu::permute_2D_sparse_data_cuda);
   DISPATCH_TO_CUDA(
       "permute_2D_sparse_data", fbgemm_gpu::permute_2D_sparse_data_cuda);
   DISPATCH_TO_CUDA(