lint

apache · Sep 12, 2022 · 09ed7ab · 09ed7ab
1 parent b2e3ad1
commit 09ed7ab
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 5 deletions.
diff --git a/src/tir/schedule/primitive/pad_einsum.cc b/src/tir/schedule/primitive/pad_einsum.cc
@@ -110,7 +110,7 @@ bool CheckTrivialBufferIndices(const T& buffer_access, Array<Var>* indices = nul
 
 class EinsumExtractor : public ExprVisitor {
  public:
-  explicit EinsumExtractor() = default;
+  EinsumExtractor() = default;
 
   std::optional<Einsum> Extract(const Block& block) {
     const BufferStoreNode* update = block->body.as<BufferStoreNode>();
@@ -493,4 +493,4 @@ struct PadEinsumTraits : public UnpackedInstTraits<PadEinsumTraits> {
 TVM_REGISTER_INST_KIND_TRAITS(PadEinsumTraits);
 
 }  // namespace tir
-}  // namespace tvm
+}  // namespace tvm
diff --git a/src/tir/schedule/transform.h b/src/tir/schedule/transform.h
@@ -114,8 +114,7 @@ class ReplaceBufferMutator : public StmtExprMutator {
   ReplaceBufferMutator(const Buffer& old_buffer, Buffer new_buffer,
                        Map<Block, Block>* block_sref_reuse);
 
-  ReplaceBufferMutator(const Map<Buffer, Buffer>& buffer_map,
-                       Map<Block, Block>* block_sref_reuse);
+  ReplaceBufferMutator(const Map<Buffer, Buffer>& buffer_map, Map<Block, Block>* block_sref_reuse);
 
  protected:
   using StmtExprMutator::VisitExpr_;

diff --git a/tests/python/unittest/test_tir_schedule_pad_einsum.py b/tests/python/unittest/test_tir_schedule_pad_einsum.py
@@ -106,7 +106,6 @@ def test_pad_matmul():
     sch = tir.Schedule(matmul_before, debug_mask="all")
     C = sch.get_block("C_shared")
     sch.pad_einsum(C, [0, 1, 1])
-    print(sch.mod["main"].script())
     tvm.ir.assert_structural_equal(matmul_expected, sch.mod["main"])
     verify_trace_roundtrip(sch, mod=matmul_before)