This is an automated email from the ASF dual-hosted git repository.
tqchen pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/incubator-tvm.git
The following commit(s) were added to refs/heads/master by this push:
new 9d1d8a9 [PYTHON] Add buffer name when creating tensor bindings (#5670)
9d1d8a9 is described below
commit 9d1d8a90ff7c25355d8098342c55f7e7c198282c
Author: Tianqi Chen <[email protected]>
AuthorDate: Mon May 25 20:19:15 2020 -0700
[PYTHON] Add buffer name when creating tensor bindings (#5670)
---
python/tvm/driver/build_module.py | 1 +
1 file changed, 1 insertion(+)
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)
