From 5b37c413cb4b7150c7611d820c5f5445b84c0a2a Mon Sep 17 00:00:00 2001 From: notoraptor Date: Fri, 29 May 2020 15:52:12 -0400 Subject: [PATCH] In memory_plan, check if value is not None, instead of just checking value as boolean. --- python/tvm/relay/transform/memory_plan.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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: