Skip to content

Commit

Permalink
disable other rewrite to test CI (apache#7371)
Browse files Browse the repository at this point in the history
  • Loading branch information
masahi authored and trevor-m committed Mar 2, 2021
1 parent 4a20df5 commit 1049d58
Showing 1 changed file with 7 additions and 5 deletions.
12 changes: 7 additions & 5 deletions tests/python/frontend/pytorch/test_object_detection.py
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand All @@ -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
Expand Down

0 comments on commit 1049d58

Please sign in to comment.