try run dot product schedule on CI

apache · Mar 31, 2022 · 53ff53e · 53ff53e
1 parent b6729a5
commit 53ff53e
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/tests/python/topi/python/test_topi_conv2d_int8.py b/tests/python/topi/python/test_topi_conv2d_int8.py
@@ -363,19 +363,19 @@ def get_ref_data():
         # ),
     ]
 
-    # TODO(tvm-team): Figure out ARM dot product availability on CI aarch64 environment
+    build_only_aarch64 = platform.machine() != "aarch64"
+
     targets.append(
         (
             "llvm -device arm_cpu -mtriple aarch64-linux-gnu -mattr=+neon,+v8.2a,+dotprod",
             topi.arm_cpu.conv2d_NCHWc_int8,
             topi.arm_cpu.schedule_conv2d_NCHWc_int8,
             8,
-            True,
+            build_only_aarch64,
         )
     )
 
     if in_dtype == "int8":
-        build_only_aarch64 = platform.machine() != "aarch64"
         targets.append(
             (
                 "llvm -device arm_cpu -mtriple aarch64-linux-gnu -mattr=+neon",