add warning when storage align doesn't work

apache · Jul 13, 2022 · 5ad0386 · 5ad0386
1 parent 8b7fc70
commit 5ad0386
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/src/meta_schedule/schedule_rule/multi_level_tiling_tensor_core.cc b/src/meta_schedule/schedule_rule/multi_level_tiling_tensor_core.cc
@@ -227,6 +227,9 @@ std::vector<State> MultiLevelTilingTensorCoreNode::AddReadReuseTensorCore(
       sch->StorageAlign(cache_read, 0, -2, 32, 8);
     } else if (dtype.is_int() && dtype.bits() == 8) {
       sch->StorageAlign(cache_read, 0, -2, 32, 16);
+    } else {
+      LOG(WARNING) << "StorageAlign is not applied for data type " << dtype
+                   << ", shared memory accesses might be inefficient.";
     }
   }
   return {state};

diff --git a/src/tir/schedule/analysis.h b/src/tir/schedule/analysis.h
@@ -423,7 +423,7 @@ struct ProducerConsumerSplit {
  * \param self The schedule state.
  * \param block The queried block.
  * \param n The index of the queried buffer.
- * \param buffer_index_type The type of the buffer index, kRead or kWrite.
+ * \param index_type The type of the buffer index, kRead or kWrite.
  * \return The buffer of the n-th read/write region of the block.
  * \throw ScheduleError If the buffer index is out of bound.
  */
@@ -435,7 +435,7 @@ Buffer GetNthAccessBuffer(const ScheduleState& self, const Block& block, int n,
  * \param self The schedule state.
  * \param block The queried block.
  * \param n The index of the queried buffer.
- * \param buffer_index_type The type of the buffer index, kRead or kWrite.
+ * \param index_type The type of the buffer index, kRead or kWrite.
  * \return The n-th read/write region of the block.
  * \throw ScheduleError If the buffer index is out of bound.
  */