mbs-octoml commented on a change in pull request #9689:
URL: https://github.com/apache/tvm/pull/9689#discussion_r766111536
##########
File path: python/tvm/tir/analysis/analysis.py
##########
@@ -196,3 +197,70 @@ def detect_buffer_access_lca(func: PrimFunc) ->
Dict[Buffer, Stmt]:
Map from buffer to the LCA of all access to it.
"""
return _ffi_api.detect_buffer_access_lca(func) # type: ignore # pylint:
disable=no-member
+
+
+# NOTE: relay_func_type in the following two functions should be
relay.FuncType however that would
+# introduce a cycling dependency. We make do with Object.
+
+
+def get_prim_func_arg_and_result_memory_constraints(
+ func: PrimFunc, relay_func_type: Object
+) -> List[AnyStr]:
Review comment:
only my ignorance of 'str' as the type :-) fixed
--
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: [email protected]
For queries about this service, please contact Infrastructure at:
[email protected]
