This is an automated email from the ASF dual-hosted git repository.
tqchen pushed a commit to branch script/canonical-parser-df
in repository https://gitbox.apache.org/repos/asf/tvm.git
The following commit(s) were added to refs/heads/script/canonical-parser-df by
this push:
new e62f985078 Declare VNNI tensorization reference storage explicitly
e62f985078 is described below
commit e62f9850784d4075fecb39ebcc61b90b2f17e0e9
Author: Tianqi Chen <[email protected]>
AuthorDate: Wed Sep 23 10:12:38 2026 +0000
Declare VNNI tensorization reference storage explicitly
---
.../test_meta_schedule_postproc_rewrite_tensorize.py | 10 +++++-----
1 file changed, 5 insertions(+), 5 deletions(-)
diff --git
a/tests/python/s_tir/meta_schedule/test_meta_schedule_postproc_rewrite_tensorize.py
b/tests/python/s_tir/meta_schedule/test_meta_schedule_postproc_rewrite_tensorize.py
index 57c345ef30..2d0fa4d034 100644
---
a/tests/python/s_tir/meta_schedule/test_meta_schedule_postproc_rewrite_tensorize.py
+++
b/tests/python/s_tir/meta_schedule/test_meta_schedule_postproc_rewrite_tensorize.py
@@ -230,11 +230,11 @@ class Conv2dNCHWcVNNIModuleTensorized:
dtype="int32",
offset_factor=1,
)
- A_u8x4 = A.vload([0], "uint8x4")
- A_i32 = T.reinterpret(A_u8x4, dtype="int32")
- B_i8x64 = B.vload([0, 0], dtype="int8x64")
- B_i32x16 = T.reinterpret(B_i8x64, dtype="int32x16")
- C_i32x16 = C.vload([0], dtype="int32x16")
+ A_u8x4: T.uint8x4 = A.vload([0], "uint8x4")
+ A_i32: T.int32 = T.reinterpret(A_u8x4, dtype="int32")
+ B_i8x64: T.int8x64 = B.vload([0, 0], dtype="int8x64")
+ B_i32x16: T.int32x16 = T.reinterpret(B_i8x64,
dtype="int32x16")
+ C_i32x16: T.int32x16 = C.vload([0], dtype="int32x16")
C[T.ramp(0, 1, 16)] = T.call_llvm_pure_intrin(
T.llvm_lookup_intrinsic_id("llvm.x86.avx512.vpdpbusd.512"),
C_i32x16,
