Skip to content

Commit 495606d

Browse files
committed
Fix more tests
1 parent b3331bc commit 495606d

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

tests/python/relay/test_pass_combine_parallel_dense.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -29,6 +29,7 @@ def run_combine_parallel(expr, min_num_branches=3, to_batch=True):
2929
def run_opt_pass(expr, opt_pass):
3030
assert isinstance(opt_pass, tvm.transform.Pass)
3131
mod = tvm.IRModule.from_expr(expr)
32+
mod = tvm.relay.transform.InferType()(mod)
3233
mod = opt_pass(mod)
3334
return mod["main"]
3435

0 commit comments

Comments
 (0)