From e4c51591aad62acf678a77c261cd23aa73a6cc8c Mon Sep 17 00:00:00 2001 From: Ruihang Lai Date: Tue, 28 May 2024 21:15:51 -0400 Subject: [PATCH] [HotFix] Pylint in python/tvm/topi/arm_cpu/conv2d.py --- python/tvm/topi/arm_cpu/conv2d.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/python/tvm/topi/arm_cpu/conv2d.py b/python/tvm/topi/arm_cpu/conv2d.py index d0fe251e7e..6b916e8e7a 100644 --- 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