disable one of rewrite in torch detection test (#7365)

apache · Jan 29, 2021 · 02c764c · 02c764c
1 parent 4142128
commit 02c764c
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/tests/python/frontend/pytorch/test_object_detection.py b/tests/python/frontend/pytorch/test_object_detection.py
@@ -150,10 +150,10 @@ def compile_and_run_vm(mod, params, data_np, target):
     after = mod["main"]
     assert not tvm.ir.structural_equal(after, before)
 
-    before = mod["main"]
-    mod = rewrite_scatter_to_gather(mod, 4)  # num_scales is 4 for maskrcnn_resnet50_fpn
-    after = mod["main"]
-    assert not tvm.ir.structural_equal(after, before)
+    # before = mod["main"]
+    # mod = rewrite_scatter_to_gather(mod, 4)  # num_scales is 4 for maskrcnn_resnet50_fpn
+    # after = mod["main"]
+    # assert not tvm.ir.structural_equal(after, before)
 
     tvm_res_after_rewrite = compile_and_run_vm(mod, params, data_np, "llvm")