[HotFix] Pylint in python/tvm/topi/arm_cpu/conv2d.py

mlc-ai · May 31, 2024 · e4c5159 · e4c5159
1 parent 1c12efc
commit e4c5159
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/python/tvm/topi/arm_cpu/conv2d.py b/python/tvm/topi/arm_cpu/conv2d.py
@@ -758,11 +758,13 @@ def schedule_conv2d_NHWC_hybrid_TIR(sch: tvm.tir.Schedule):
 
         # Tensorize the GeMM update
         sme_gemm_interleaved_intrin_name = ARM_SME_2SVLx2SVL_GEMM_INTERLEAVED_MOPA + f"_{K_padded}"
+        # pylint: disable=no-value-for-parameter
         tvm.tir.TensorIntrin.register(
             sme_gemm_interleaved_intrin_name,
             *get_sme_gemm_interleaved_mopa_2svlx2svl_intrin(K_padded, dtype),
             override=True,
         )
+        # pylint: enable=no-value-for-parameter
         sch.tensorize(mi, sme_gemm_interleaved_intrin_name)
 
         # Add pstate annotations