Add simplify pass in the Relay interpreter (#2417)
This commit is contained in:
Родитель
547a0913f6
Коммит
312802f341
|
@ -218,7 +218,9 @@ class Interpreter(Executor):
|
|||
"""
|
||||
# TODO: We need to move this optimization code into the optimizer/pass manager
|
||||
ck_expr = ir_pass.infer_type(expr, mod=self.mod)
|
||||
fused_expr = ir_pass.fuse_ops(ck_expr)
|
||||
simp_expr = ir_pass.simplify_inference(ck_expr)
|
||||
ck_simp = ir_pass.infer_type(simp_expr, mod=self.mod)
|
||||
fused_expr = ir_pass.fuse_ops(ck_simp)
|
||||
ck_fused = ir_pass.infer_type(fused_expr, mod=self.mod)
|
||||
return ck_fused
|
||||
|
||||
|
|
Загрузка…
Ссылка в новой задаче