commit 91a72e9021c73b19b2db2a72e65e53e514c8abe0
author Ansha Yu <ansha@fb.com> Tue Mar 29 11:25:35 2022 -0700
committer PyTorch MergeBot <pytorchmergebot@users.noreply.github.com> Tue Mar 29 18:30:32 2022 +0000
tree 947a9f6960eb5f3ce310eab39c060a26f5ed46b9
parent 1249d490de6967465050dd94238d7470e34b3951

Back out D34696255 "[pyper] to + lengths_to_offsets" (#74906)

Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/74906

Original commit changeset: a0dc4a8ff8f2

Original Phabricator Diff: D34696255 (https://github.com/pytorch/pytorch/commit/f2ca4341c9008472022227b43d943af3e5e46e81)

Test Plan: New models published can't be served on nnpi

Reviewed By: mikeiovine

Differential Revision: D35214444

fbshipit-source-id: 5691ded93d76074bbe5fe78bb12519389360a69e
(cherry picked from commit 8357a13fd8ffe55bf3577fb13b9b39436e0b67f7)

torch/csrc/jit/runtime/static/passes.cpp

diff --git a/torch/csrc/jit/runtime/static/passes.cpp b/torch/csrc/jit/runtime/static/passes.cpp
index 7dc35cd..b05b8ab 100644
--- a/torch/csrc/jit/runtime/static/passes.cpp
+++ b/torch/csrc/jit/runtime/static/passes.cpp
@@ -226,33 +226,6 @@
   fuse.runOnGraph(graph);
 }
 
-C10_UNUSED void ToLengthsToOffsets(std::shared_ptr<torch::jit::Graph>& graph) {
-  std::string pattern = R"IR(
-    graph(%a, %includelastoffset, %dtype, %nonblocking, %copy, %memoryformat):
-        %y0 : Tensor = aten::to(%a, %dtype, %nonblocking, %copy, %memoryformat)
-        %y1 : Tensor = fb::lengths_to_offsets(%y0, %includelastoffset)
-        return (%y1))IR";
-  std::string fused_pattern = R"IR(
-    graph(%a, %includelastoffset, %dtype, %nonblocking, %copy, %memoryformat):
-        %y0 : Tensor = fb::to_lengths_to_offsets(%a, %includelastoffset, %dtype)
-        return (%y0))IR";
-  SubgraphRewriter fuse;
-  fuse.RegisterRewritePattern(pattern, fused_pattern);
-  fuse.runOnGraph(graph);
-
-  std::string pattern2 = R"IR(
-    graph(%a, %includelastoffset, %dtype, %nonblocking, %copy):
-        %y0 : Tensor = aten::to(%a, %dtype, %nonblocking, %copy)
-        %y1 : Tensor = fb::lengths_to_offsets(%y0, %includelastoffset)
-        return (%y1))IR";
-  std::string fused_pattern2 = R"IR(
-    graph(%a, %includelastoffset, %dtype, %nonblocking, %copy):
-        %y0 : Tensor = fb::to_lengths_to_offsets(%a, %includelastoffset, %dtype)
-        return (%y0))IR";
-  fuse.RegisterRewritePattern(pattern2, fused_pattern2);
-  fuse.runOnGraph(graph);
-}
-
 C10_UNUSED
 void ClipRangesGatherSigridHash(std::shared_ptr<torch::jit::Graph>& graph) {
   // TODO:: check restrictions for inputs; outputs not used elsewhere
@@ -362,8 +335,6 @@
 
     ClipRangesToGatherToOffsets(graph);
   }
-
-  ToLengthsToOffsets(graph);
 #endif
 }