From 627d439e753a9aa533c68e3161043e21c4133304 Mon Sep 17 00:00:00 2001 From: masahi Date: Wed, 10 Aug 2022 14:17:17 +0900 Subject: [PATCH] [TIR] Minor fix to tensor intrin description (#12356) --- python/tvm/tir/tensor_intrin/arm_cpu.py | 2 -- python/tvm/tir/tensor_intrin/x86.py | 2 -- 2 files changed, 4 deletions(-) diff --git a/python/tvm/tir/tensor_intrin/arm_cpu.py b/python/tvm/tir/tensor_intrin/arm_cpu.py index 6e16b1f767f3..3e934e1b9d2d 100644 --- a/python/tvm/tir/tensor_intrin/arm_cpu.py +++ b/python/tvm/tir/tensor_intrin/arm_cpu.py @@ -34,8 +34,6 @@ def dot_product_4x4_i8i8i32_desc( T.reads(C[0:4], A[0:4], B[0:4, 0:4]) T.writes(C[0:4]) for i in T.serial(0, 4): - with T.init(): - C[i] = T.int32(0) for k in T.serial(0, 4): with T.block("update"): vi, vk = T.axis.remap("SR", [i, k]) diff --git a/python/tvm/tir/tensor_intrin/x86.py b/python/tvm/tir/tensor_intrin/x86.py index ee57c9aa4750..f1c603228715 100644 --- a/python/tvm/tir/tensor_intrin/x86.py +++ b/python/tvm/tir/tensor_intrin/x86.py @@ -34,8 +34,6 @@ def dot_product_16x4_u8i8i32_desc( T.reads(C[0:16], A[0:4], B[0:16, 0:4]) T.writes(C[0:16]) for i in T.serial(0, 16): - with T.init(): - C[i] = T.int32(0) for k in T.serial(0, 4): with T.block("update"): vi, vk = T.axis.remap("SR", [i, k])