Skip to content

Commit

Permalink
[TIR] Minor fix to tensor intrin description (apache#12356)
Browse files Browse the repository at this point in the history
  • Loading branch information
masahi authored and Mikael Sevenier committed Aug 12, 2022
1 parent d8c904a commit 627d439
Show file tree
Hide file tree
Showing 2 changed files with 0 additions and 4 deletions.
2 changes: 0 additions & 2 deletions python/tvm/tir/tensor_intrin/arm_cpu.py
Original file line number Diff line number Diff line change
Expand Up @@ -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])
Expand Down
2 changes: 0 additions & 2 deletions python/tvm/tir/tensor_intrin/x86.py
Original file line number Diff line number Diff line change
Expand Up @@ -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])
Expand Down

0 comments on commit 627d439

Please sign in to comment.