[tvm] 01/01: fix assert

[an...@apache.org]

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 <an...@gmail.com>
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,