diff --git a/vta/python/vta/transform.py b/vta/python/vta/transform.py index b1135c0eb0070..5a26ecf83af74 100644 --- a/vta/python/vta/transform.py +++ b/vta/python/vta/transform.py @@ -202,7 +202,14 @@ def _post_order(op): ), op.body, ) - alloc = tvm.tir.Allocate(buffer_var, op.dtype, op.extents, op.condition, let_stmt) + alloc = tvm.tir.Allocate( + buffer_var, + op.dtype, + op.extents, + op.condition, + let_stmt, + annotations={"disable_lower_builtin": True}, + ) del var_remap[buffer_var] bufs_to_delete = [ old_buf for old_buf in buf_remap if old_buf.data.same_as(buffer_var)