Pass GlobalVarSupply when lowering shape funcs

apache · Jul 18, 2022 · 4ccc004 · 4ccc004
1 parent 3c327fe
commit 4ccc004
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/src/relay/backend/te_compiler.cc b/src/relay/backend/te_compiler.cc
@@ -499,8 +499,7 @@ class TECompilerImpl : public TECompilerNode {
 
     using tvm::transform::PassContext;
     With<PassContext> fresh_pass_ctx_scope(PassContext::Create());
-    value->cached_func =
-        ShapeFuncFor(key->source_func, key->target, GlobalVarSupply::EmptySupply());
+    value->cached_func = ShapeFuncFor(key->source_func, key->target, global_var_supply);
 
     ICHECK(
         value->cached_func->funcs->Lookup(value->cached_func->prim_fn_var).as<tir::PrimFuncNode>());