[FIX][TIR] Remove unused code and fix typo in storage_align (#9583)

apache · Nov 25, 2021 · 0754c86 · 0754c86
1 parent d6fe9a3
commit 0754c86
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 2 deletions.
diff --git a/python/tvm/tir/schedule/schedule.py b/python/tvm/tir/schedule/schedule.py
@@ -1631,7 +1631,7 @@ def before_storage_align(a: T.handle, c: T.handle) -> None:
             sch.storage_align(sch.get_block("B"), buffer_index=0, axis=0, factor=128, offset=1)
             print(sch.mod["main"].script())
 
-        After applying rfactor, the IR becomes:
+        After applying storage_align, the IR becomes:
 
         .. code-block:: python
 

diff --git a/src/tir/schedule/primitive/block_annotate.cc b/src/tir/schedule/primitive/block_annotate.cc
@@ -248,7 +248,6 @@ void StorageAlign(ScheduleState self, const StmtSRef& block_sref, int buffer_ind
                                                                 GetRef<Block>(block_ptr));
 
   // Step 2: Update the annotation value
-  // Array<Array<Integer>> buffer_storage_align = storage_align_annotation[buffer_index];
   bool found = false;
   StorageAlignTuple new_storage_align_tuple{Integer(buffer_index), Integer(axis), Integer(factor),
                                             Integer(offset)};