fix lint

Author: Siyuan Feng

python/tvm/tir/schedule/schedule.py

@@ -853,7 +853,9 @@ def after_cache_read(a: ty.handle, b: ty.handle) -> None:
                         B[vi, vj] = A_local[vi, vj] * 2.0
 
         """
-        return _ffi_api.ScheduleCacheRead(self, block, read_buffer_index, storage_scope)  # type: ignore # pylint: disable=no-member
+        return _ffi_api.ScheduleCacheRead(  # type: ignore # pylint: disable=no-member
+            self, block, read_buffer_index, storage_scope
+        )
 
     def cache_write(self, block: BlockRV, write_buffer_index: int, storage_scope: str) -> BlockRV:
         """Create a block that reads a buffer region into a write cache. It requires:
@@ -919,7 +921,9 @@ def after_cache_write(a: ty.handle, b: ty.handle) -> None:
                         B[vi, vj] = B_local[vi, vj]
 
         """
-        return _ffi_api.ScheduleCacheWrite(self, block, write_buffer_index, storage_scope)  # type: ignore # pylint: disable=no-member
+        return _ffi_api.ScheduleCacheWrite(  # type: ignore # pylint: disable=no-member
+            self, block, write_buffer_index, storage_scope
+        )
 
     ########## Schedule: Compute location ##########
 

src/tir/schedule/analysis/analysis.cc

@@ -634,7 +634,7 @@ Buffer GetNthAccessBuffer(const ScheduleState& self, const Block& block, int n,
 
   const Array<BufferRegion>& access_region = is_write ? block->writes : block->reads;
 
-  if (n < 0 || access_region.size() <= n) {
+  if (n < 0 || static_cast<int>(access_region.size()) <= n) {
     throw BufferIndexOutOfRangeError(self->mod, block, n, is_write);
   }
   return access_region[n]->buffer;

src/tir/schedule/primitive/cache_read_write.cc

@@ -185,7 +185,8 @@ SeqStmt InsertCacheStage(const Stmt& stmt, int pos, const Stmt& stage) {
  * \param self The state of the schedule
  * \param scope_sref The scope block where the write is considered
  * \param buffer The queried buffer
- * \return The sref of the only writer of the input buffer in the given scope, or `NullOpt` if no block writes it in the scope.
+ * \return The sref of the only writer of the input buffer in the given scope,
+ *         or `NullOpt` if no block writes it in the scope.
  * \throw NotSingleWriteBlock if there are more than one intrested block.
  */
 Optional<StmtSRef> GetOnlyWriteBlock(ScheduleState self, const StmtSRef& scope_sref,
@@ -239,11 +240,10 @@ BufferRegion RelaxBufferRegion(ScheduleState self, const BufferRegion& buffer_re
 class CacheLocDetector : public StmtVisitor {
  public:
   /*!
-   * \brief Detect the insertion position of the cache stage, and write the position into the CacheStageInfo
-   * \param self The state of the schedule
-   * \param block_sref The sref of the unique writer block of the buffer being applied cache_read or cache_write
-   * \param scope_sref The sref of the scope block of the cached block
-   * \param info The cache stage info.
+   * \brief Detect the insertion position of the cache stage, and write the position into the
+   * CacheStageInfo \param self The state of the schedule \param block_sref The sref of the unique
+   * writer block of the buffer being applied cache_read or cache_write \param scope_sref The sref
+   * of the scope block of the cached block \param info The cache stage info.
    */
   static void Detect(const ScheduleState& self, const StmtSRef& block_sref,
                      const StmtSRef& scope_sref, CacheStageInfo* info) {
@@ -269,9 +269,9 @@ class CacheLocDetector : public StmtVisitor {
   /*!
    * \brief Constructor
    * \param self The state of the schedule
-   * \param block_sref The sref of the unique writer block of the buffer being applied cache_read or cache_write
-   * \param scope_sref The sref of the scope block of the cached block
-   * \param related_blocks Producer blocks for cache_write, or consumer blocks for cache_read
+   * \param block_sref The sref of the unique writer block of the buffer being applied cache_read or
+   * cache_write \param scope_sref The sref of the scope block of the cached block \param
+   * related_blocks Producer blocks for cache_write, or consumer blocks for cache_read
    */
   CacheLocDetector(const ScheduleState self, const StmtSRef& block_sref, const StmtSRef& scope_sref,
                    const std::vector<StmtSRef>& related_blocks)

src/tir/schedule/state.cc

@@ -1031,8 +1031,7 @@ TVM_DLL Array<Bool> GetCachedFlags(const ScheduleState& self, const StmtSRef& bl
 
 TVM_DLL void ScheduleStateNode::UpdateAffineFlag(const StmtSRef& scope_sref) {
   auto it = this->block_info.find(scope_sref);
-  ICHECK(it != this->block_info.end())
-      << "Cannot find the block info of the given block.";
+  ICHECK(it != this->block_info.end()) << "Cannot find the block info of the given block.";
   BlockInfo& info = it->second;
 
   bool is_root_block = scope_sref->parent == nullptr;