add note for dynamic shapes

Change-Id: If6409e2953addfc880bcc6d95083b78bdf5a23d0
apache · Jun 24, 2022 · 89f7523 · 89f7523
1 parent 93c0672
commit 89f7523
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/include/tvm/relay/transform.h b/include/tvm/relay/transform.h
@@ -563,6 +563,9 @@ TVM_DLL Pass FlattenAtrousConv();
  * function in question as a list of the number of bytes for each callsite. In addition, the
  * containing function is annotated with an "io_used_memory" annotation which refers to the total
  * memory required for the IO tensors.
+ *
+ * Note: This pass does not support dynamic shapes, it is the users responsibility to check this
+ * pass isn't applied where dynamic shapes may be input.
  */
 TVM_DLL Pass AnnotateUsedMemory();
 

diff --git a/src/relay/backend/annotate_used_memory.cc b/src/relay/backend/annotate_used_memory.cc
@@ -46,6 +46,9 @@ namespace backend {
  * containing function is annotated with an "io_used_memory" annotation which refers to the total
  * memory required for the IO tensors.
  *
+ * Note: This pass does not support dynamic shapes, it is the users responsibility to check this
+ * pass isn't applied where dynamic shapes may be input.
+ *
  * A simple example:
  *
  * Before: