From 7179997b9d618e88fdb0a9a592eb350772f1cc72 Mon Sep 17 00:00:00 2001 From: Masahiro Masuda Date: Sat, 30 Jan 2021 00:08:29 +0900 Subject: [PATCH] disable other rewrite to test CI --- .../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 3c94b0b846d8..a404a88393bc 100644 --- a/tests/python/frontend/pytorch/test_object_detection.py +++ b/tests/python/frontend/pytorch/test_object_detection.py @@ -122,7 +122,7 @@ def compile_and_run_vm(mod, params, data_np, target): 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 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_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