diff --git a/python/tvm/relay/transform/memory_plan.py b/python/tvm/relay/transform/memory_plan.py index a6c2c11745ed..e359a9e3def0 100644 --- a/python/tvm/relay/transform/memory_plan.py +++ b/python/tvm/relay/transform/memory_plan.py @@ -243,7 +243,7 @@ def mk_let(self, dynamic_regions): def _mk_let(bindings, body): for var, value in reversed(bindings): assert var - assert value + assert value is not None assert body body = expr.Let(var, value, body) if var in dynamic_regions: