This is an automated email from the ASF dual-hosted git repository. andrewzhaoluo pushed a commit to branch aluo/fix-int8-tensorcore in repository https://gitbox.apache.org/repos/asf/tvm.git
commit 7eab14aace2c76cda9c4c1ee99e4c40f87c5d81c Author: Andrew Zhao Luo <[email protected]> AuthorDate: Wed Aug 10 13:31:44 2022 -0700 fix assert --- python/tvm/tir/tensor_intrin/cuda.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/python/tvm/tir/tensor_intrin/cuda.py b/python/tvm/tir/tensor_intrin/cuda.py index 4ac9338ba8..b4f5d1d331 100644 --- a/python/tvm/tir/tensor_intrin/cuda.py +++ b/python/tvm/tir/tensor_intrin/cuda.py @@ -883,8 +883,8 @@ def get_wmma_intrin_group( A group of tensor intrinsics. """ assert store_scope in ["global", "shared"] - assert in_dtype in ["float16"] - assert out_dtype in ["float16", "float32"] + assert in_dtype in ["float16", "int8"] + assert out_dtype in ["float16", "float32", "int32"] load_a_intrins = { "float16": WMMA_LOAD_16x16x16_F16_A_INTRIN,
