rollback part of D28312997 (pytorch#611)

Summary: Pull Request resolved: pytorch#611 Reviewed By: xw285cornell Differential Revision: D28541274 fbshipit-source-id: 00d5e2889a34d34b3d42c2a4f9cc81ba4ec17247
joebos · May 19, 2021 · 442fb1d · 442fb1d
1 parent d4a6222
commit 442fb1d
Show file tree

Hide file tree

Showing 2 changed files with 0 additions and 3 deletions.
diff --git a/fbgemm_gpu/src/sparse_ops_cpu.cpp b/fbgemm_gpu/src/sparse_ops_cpu.cpp
@@ -246,7 +246,6 @@ std::tuple<Tensor, Tensor, c10::optional<Tensor>> permute_sparse_data_cpu(
 } // namespace at
 
 TORCH_LIBRARY_FRAGMENT(fb, m) {
-  m.def("asynchronous_exclusive_cumsum(Tensor t_in) -> Tensor");
   m.def(
       "permute_sparse_data(Tensor permute, Tensor lengths, Tensor values, Tensor? weights=None) -> (Tensor, Tensor, Tensor?)");
 }

diff --git a/fbgemm_gpu/src/sparse_ops_gpu.cpp b/fbgemm_gpu/src/sparse_ops_gpu.cpp
@@ -13,7 +13,5 @@
 #include <torch/library.h>
 
 TORCH_LIBRARY_IMPL(fb, CUDA, m) {
-  DISPATCH_TO_CUDA(
-      "asynchronous_exclusive_cumsum", at::asynchronous_exclusive_cumsum);
   DISPATCH_TO_CUDA("permute_sparse_data", at::permute_sparse_data_cuda);
 }