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