check transpose condition

Author: masahi

python/tvm/topi/x86/batch_matmul.py

@@ -120,7 +120,9 @@ def batch_matmul(
     mcpu = tvm.target.Target.current().mcpu
 
     if (
-        target_has_vnni(mcpu)
+        transpose_a == False
+        and transpose_b == True
+        and target_has_vnni(mcpu)
         and tensor_a.dtype == "uint8"
         and tensor_b.dtype == "int8"
         and tensor_b.shape[-2] % 16 == 0

tests/python/relay/test_op_level10.py

@@ -408,6 +408,9 @@ def test_batch_matmul_vnni():
     with tvm.transform.PassContext(opt_level=3):
         lib = relay.build(mod, target=target)
 
+    asm = lib.lib.get_source("asm")
+    assert "vpdpbusd" in asm
+
     dev = tvm.device(target, 0)
     runtime = tvm.contrib.graph_executor.GraphModule(lib["default"](dev))