From 439d59db3a3e679334b048cb82c909fc3b4403d8 Mon Sep 17 00:00:00 2001 From: Zhi Chen Date: Sat, 18 Apr 2020 16:39:26 +0000 Subject: [PATCH] alloc_storage --- python/tvm/relay/transform/memory_alloc.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/tvm/relay/transform/memory_alloc.py b/python/tvm/relay/transform/memory_alloc.py index 5fb4909b57c7..e05befcc70dd 100644 --- a/python/tvm/relay/transform/memory_alloc.py +++ b/python/tvm/relay/transform/memory_alloc.py @@ -174,7 +174,7 @@ def dynamic_invoke(self, scope, func, ins, new_args, out_types, ret_type): size = self.compute_storage_in_relay( out_shape, out_type.dtype) alignment = self.compute_alignment(out_type.dtype) - sto = scope.let("storage_{i}".format(i=i), self.alloc_storage( + sto = scope.let("storage_{i}".format(i=i), alloc_storage( size, alignment, self.default_context, out_type.dtype)) storages.append(sto)