diff --git a/python/tvm/tir/tensor_intrin/arm_cpu.py b/python/tvm/tir/tensor_intrin/arm_cpu.py index 6aebc5be3417..a360bf609b18 100644 --- a/python/tvm/tir/tensor_intrin/arm_cpu.py +++ b/python/tvm/tir/tensor_intrin/arm_cpu.py @@ -23,6 +23,7 @@ # TODO(masahi): Parametrize the TVMScript description of dot product by # shape and dtype, and share the common description with x86. + @T.prim_func def dot_product_4x4_i8i8i32_desc( A: T.Buffer((4,), "int8", offset_factor=1),