We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent b3331bc commit 495606dCopy full SHA for 495606d
tests/python/relay/test_pass_combine_parallel_dense.py
@@ -29,6 +29,7 @@ def run_combine_parallel(expr, min_num_branches=3, to_batch=True):
29
def run_opt_pass(expr, opt_pass):
30
assert isinstance(opt_pass, tvm.transform.Pass)
31
mod = tvm.IRModule.from_expr(expr)
32
+ mod = tvm.relay.transform.InferType()(mod)
33
mod = opt_pass(mod)
34
return mod["main"]
35
0 commit comments