Qianshui-Jiang commented on code in PR #13642:
URL: https://github.com/apache/tvm/pull/13642#discussion_r1056955524
##########
python/tvm/topi/x86/dense_alter_op.py:
##########
@@ -25,13 +25,15 @@
from ..utils import get_const_tuple
from ..nn import dense_alter_layout
from .utils import target_has_vnni
+from .utils import target_has_amx
from .. import nn
-def check_vnni_applicable(x, y, allow_padding=False):
+def check_inst_applicable(x, y, allow_padding=False):
mcpu = tvm.target.Target.current().mcpu
+ cpu_avai = target_has_vnni(mcpu) or target_has_amx(mcpu)
return (
- target_has_vnni(mcpu)
+ cpu_avai
Review Comment:
yep, replaced `cpu_avai` to `simd_avai` to check if vnni or amx was
availiable in this target.
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
To unsubscribe, e-mail: [email protected]
For queries about this service, please contact Infrastructure at:
[email protected]
