[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 (file)
--- 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)