From 015960682c8153ca739c955f296482af1025740b Mon Sep 17 00:00:00 2001 From: Matthew Brookhart Date: Thu, 13 Jan 2022 20:02:55 -0700 Subject: [PATCH] fix icelake target for avx512 and vnni (#9928) --- python/tvm/topi/x86/utils.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/python/tvm/topi/x86/utils.py b/python/tvm/topi/x86/utils.py index 658a92966257b..50c5c848ee0aa 100644 --- a/python/tvm/topi/x86/utils.py +++ b/python/tvm/topi/x86/utils.py @@ -81,8 +81,8 @@ def target_has_avx512(target): # explicit enumeration of VNNI capable due to collision with alderlake "cascadelake", "icelake-client", + "icelake-server", "rocketlake", - "icelake", "tigerlake", "cooperlake", "sapphirerapids", @@ -93,8 +93,8 @@ def target_has_vnni(target): return target in { "cascadelake", "icelake-client", + "icelake-server", "rocketlake", - "icelake", "tigerlake", "cooperlake", "sapphirerapids",