From bc5e3ec77922d8a462d7951e2638cb06ee8752cf Mon Sep 17 00:00:00 2001 From: Masahiro Masuda Date: Thu, 24 Dec 2020 05:49:56 +0900 Subject: [PATCH] removed TODO in rewrite check --- tests/python/frontend/pytorch/test_object_detection.py | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) diff --git a/tests/python/frontend/pytorch/test_object_detection.py b/tests/python/frontend/pytorch/test_object_detection.py index 2476a3e3583c2..569dce0d64a7c 100644 --- a/tests/python/frontend/pytorch/test_object_detection.py +++ b/tests/python/frontend/pytorch/test_object_detection.py @@ -26,8 +26,7 @@ from tvm import relay from tvm.runtime.vm import VirtualMachine -from tvm.relay.dataflow_pattern import rewrite -from tvm.relay.frontend.pytorch_utils import NMSRewrite +from tvm.relay.frontend.pytorch_utils import rewrite_nms_to_batched_nms from tvm.contrib.download import download @@ -142,8 +141,7 @@ def count_valid_scores(scores): ) before = mod["main"] - after = rewrite(NMSRewrite(), before) - # TODO(masahi): Is there a better way to test if the desired rewrite has happened? + after = rewrite_nms_to_batched_nms(before) assert not tvm.ir.structural_equal(after, before) mod["main"] = after