You are viewing a plain text version of this content. The canonical link for it is here.
Posted to commits@tvm.apache.org by zh...@apache.org on 2021/01/29 07:38:24 UTC
[tvm] branch main updated: disable one of rewrite in torch
detection test (#7365)
This is an automated email from the ASF dual-hosted git repository.
zhic pushed a commit to branch main
in repository https://gitbox.apache.org/repos/asf/tvm.git
The following commit(s) were added to refs/heads/main by this push:
new 02c764c disable one of rewrite in torch detection test (#7365)
02c764c is described below
commit 02c764c72f70d5f08be1385737cea706e872def0
Author: masahi <ma...@gmail.com>
AuthorDate: Fri Jan 29 16:38:09 2021 +0900
disable one of rewrite in torch detection test (#7365)
---
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 fd33dd1..6b7f9be 100644
--- a/tests/python/frontend/pytorch/test_object_detection.py
+++ b/tests/python/frontend/pytorch/test_object_detection.py
@@ -150,10 +150,10 @@ def test_detection_models():
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")