[GitHub] [tvm] Lunderberg commented on a diff in pull request #12891: [TIR][Transform] Clear buffer_map during MakeUnpackedAPI

[GitBox <gi...@apache.org>]

Lunderberg commented on code in PR #12891:
URL: https://github.com/apache/tvm/pull/12891#discussion_r981396998


##########
python/tvm/relay/backend/contrib/ethosu/tir_to_cs_translator.py:
##########
@@ -254,15 +255,19 @@ def extract_param_base_addresses(mod, buffer_info, scratch_region_map) -> List[u
     assert len(mod.functions.items()) == 1
     primfunc = mod.functions.items()[0][1]
 
+    buffer_map = tir_utils.collect_buffer_map(primfunc.body)
+
     base_addresses = list()
     idx = 0
+
     for param in primfunc.params:
         # constants are pooled together and handled specially
         # this will change after tir.allocate_const.
         # For now, we are skipping generating buffer addresses here
         if buffer_info[param].btype == BufferType.constant:
             continue
-        buffer = primfunc.buffer_map[param]
+

Review Comment:
   That works for me locally, on to the CI!  Thank you on the ethos-u debugging, and it helps quite a bit.



-- 
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.

To unsubscribe, e-mail: commits-unsubscribe@tvm.apache.org

For queries about this service, please contact Infrastructure at:
users@infra.apache.org