This is an automated email from the ASF dual-hosted git repository.
tqchen 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 96b0981 disable other rewrite to test CI (#7371)
96b0981 is described below
commit 96b09817fc1796a789524ae30cd2d7e9d6f73d6c
Author: masahi <[email protected]>
AuthorDate: Mon Feb 1 10:20:09 2021 +0900
disable other rewrite to test CI (#7371)
---
tests/python/frontend/pytorch/test_object_detection.py | 12 +++++++-----
1 file changed, 7 insertions(+), 5 deletions(-)
diff --git a/tests/python/frontend/pytorch/test_object_detection.py
b/tests/python/frontend/pytorch/test_object_detection.py
index 3c94b0b..a404a88 100644
--- a/tests/python/frontend/pytorch/test_object_detection.py
+++ b/tests/python/frontend/pytorch/test_object_detection.py
@@ -122,7 +122,7 @@ def test_detection_models():
vm.set_input("main", **{input_name: data_np})
return vm.run()
- for target in ["cuda", "llvm"]:
+ for target in ["llvm"]:
tvm_res = compile_and_run_vm(mod, params, data_np, target)
# Bounding boxes
@@ -145,10 +145,12 @@ def test_detection_models():
after = mod["main"]
assert not tvm.ir.structural_equal(after, before)
- before = mod["main"]
- mod = rewrite_batched_nms_with_max_out_size(mod)
- after = mod["main"]
- assert not tvm.ir.structural_equal(after, before)
+ # TODO(masahi): It seems this rewrite causes flaky segfaults on CI
+ # See https://github.com/apache/tvm/issues/7363
+ # before = mod["main"]
+ # mod = rewrite_batched_nms_with_max_out_size(mod)
+ # 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
