Decoupling AOT from graph memory planner

include/tvm/tir/transform.h

@@ -418,6 +418,11 @@ TVM_DLL Pass ConvertBlocksToOpaque();
  */
 TVM_DLL Pass CompactBufferAllocation();
 
+/*!
+ * This pass legalizes packed calls by wrapping their arguments into TVMValues
+ */
+TVM_DLL Pass LegalizePackedCalls();
+
 /*!
  * \brief Flatten the multi-dimensional BufferLoad and BufferStore
  *        to single dimensional Load/Store. Also remove Block to

python/tvm/tir/transform/transform.py

@@ -451,6 +451,17 @@ def LowerTVMBuiltin():
     return _ffi_api.LowerTVMBuiltin()
 
 
+def LegalizePackedCalls():
+    """Legalize packed calls to have its arguments wrapped in TVMValues
+
+    Returns
+    -------
+    fpass : tvm.transform.Pass
+        The result pass
+    """
+    return _ffi_api.LegalizePackedCalls()
+
+
 def LowerIntrin():
     """Lower target specific intrinsic calls.