From: Tianqi Chen <tqchen@users.noreply.github.com>
Date: Tue, 26 May 2020 03:19:15 +0000 (-0700)
Subject: [PYTHON] Add buffer name when creating tensor bindings (#5670)
X-Git-Tag: upstream/0.7.0~665
X-Git-Url: http://review.tizen.org/git/?a=commitdiff_plain;h=9d1d8a90ff7c25355d8098342c55f7e7c198282c;p=platform%2Fupstream%2Ftvm.git

[PYTHON] Add buffer name when creating tensor bindings (#5670)
---

diff --git a/python/tvm/driver/build_module.py b/python/tvm/driver/build_module.py
index 97ed8d8..b1c15fb 100644
--- a/python/tvm/driver/build_module.py
+++ b/python/tvm/driver/build_module.py
@@ -65,6 +65,7 @@ def get_binds(args, compact=False, binds=None):
                 buf = tvm.tir.decl_buffer(
                     x.shape,
                     dtype=x.dtype,
+                    name=x.name,
                     buffer_type=buffer_type)
                 binds[x] = buf
                 arg_list.append(buf)