From 80ababe8911eaed5b5e1f50f70dd0793afd7b95c Mon Sep 17 00:00:00 2001 From: Masahiro Masuda Date: Fri, 29 Jan 2021 07:00:54 +0900 Subject: [PATCH] disable one of rewrite in torch detection test --- tests/python/frontend/pytorch/test_object_detection.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/tests/python/frontend/pytorch/test_object_detection.py b/tests/python/frontend/pytorch/test_object_detection.py index fd33dd1da8b1..6b7f9be06d99 100644 --- 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")