From 53ff53eea8e0445d37635987e1f30d9150cfb92c Mon Sep 17 00:00:00 2001 From: Masahiro Masuda Date: Fri, 1 Apr 2022 04:17:27 +0900 Subject: [PATCH] try run dot product schedule on CI --- tests/python/topi/python/test_topi_conv2d_int8.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/tests/python/topi/python/test_topi_conv2d_int8.py b/tests/python/topi/python/test_topi_conv2d_int8.py index 8cbbabdb84de..860118531e51 100644 --- 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",