Merge pull request #974 from NVIDIA/maskrcnn

test: update the test for aten::to after fixing

Author: narendasan

tests/core/lowering/test_reduce_to_pass.cpp

@@ -6,36 +6,14 @@
 #include "torch/csrc/jit/ir/irparser.h"
 #include "torch/csrc/jit/ir/subgraph_matcher.h"
 
-TEST(LoweringPasses, ReduceToCorrectly) {
-  std::string source_graph = R"IR(
-    graph(%x, %device, %dtype, %nb, %copy, %format):
-        %out : Tensor = aten::to(%x, %device, %dtype, %nb, %copy, %format)
-        return (%out))IR";
-  std::string target_graph = R"IR(
-    graph(%x, %device, %dtype, %nb, %copy, %format):
-        %out : Tensor = aten::to(%x, %dtype, %nb, %copy, %format)
-        return (%out))IR";
-
-  torch_tensorrt::core::util::logging::get_logger().set_reportable_log_level(
-      torch_tensorrt::core::util::logging::LogLevel::kGRAPH);
-  auto sg = std::make_shared<torch::jit::Graph>();
-  torch::jit::parseIR(source_graph, &*sg);
-  torch_tensorrt::core::lowering::passes::ReduceToOperation(sg);
-
-  auto tg = std::make_shared<torch::jit::Graph>();
-  torch::jit::parseIR(target_graph, &*tg);
-
-  ASSERT_TRUE(!torch::jit::findPatternMatches(*tg, *sg).empty());
-}
-
 TEST(LoweringPasses, ReduceToDtypeLayoutCorrectly) {
   std::string source_graph = R"IR(
-    graph(%x, %device, %dtype, %layout, %pm, %nb, %copy, %format):
-        %out : Tensor = aten::to(%x, %device, %dtype, %layout, %pm, %nb, %copy, %format)
+    graph(%x, %dtype, %layout, %device, %pm, %nb, %copy, %format):
+        %out : Tensor = aten::to(%x, %dtype, %layout, %device, %pm, %nb, %copy, %format)
         return (%out))IR";
   std::string target_graph = R"IR(
-    graph(%x, %device, %dtype, %layout, %pm, %nb, %copy, %format):
-        %out : Tensor = aten::to(%x, %dtype, %nb, %copy, %format)
+    graph(%x, %dtype, %layout, %device, %pm, %nb, %copy, %format):
+        %out : Tensor = aten::to(%x, %device, %dtype, %nb, %copy, %format)
         return (%out))IR";
 
   torch_tensorrt::core::util::logging::get_logger().set_reportable_log_level(