From b282401a7bf3eb986e656c9342fb3d7b9c40aa0d Mon Sep 17 00:00:00 2001 From: Luke Hutton Date: Fri, 19 May 2023 08:48:26 +0000 Subject: [PATCH] [TEST] Enable arm_cpu qnn.conv2d i8mm test Enabling the test now we have LLVM > 11 in the docker images. Change-Id: Ifccf6ca2ebea87e57d23e660a55d1a36b2dd5d22 --- tests/python/topi/python/test_topi_conv2d_int8.py | 11 +++++------ 1 file changed, 5 insertions(+), 6 deletions(-) diff --git a/tests/python/topi/python/test_topi_conv2d_int8.py b/tests/python/topi/python/test_topi_conv2d_int8.py index e05dba3dfee4..fd101fb79768 100644 --- a/tests/python/topi/python/test_topi_conv2d_int8.py +++ b/tests/python/topi/python/test_topi_conv2d_int8.py @@ -57,12 +57,11 @@ topi.arm_cpu.compute_conv2d_NHWC_quantized_native, topi.arm_cpu.schedule_conv2d_NHWC_quantized_native, ), - # TODO(giuseros) We need LLVM-11 in order to compile with +i8mm extension - # ( - # "llvm --device arm_cpu --mtriple aarch64-linux-gnu -mattr=+v8.2a,+i8mm", - # topi.arm_cpu.compute_conv2d_NHWC_quantized_interleaved, - # topi.arm_cpu.schedule_conv2d_NHWC_quantized_interleaved, - # ), + ( + "llvm --device arm_cpu --mtriple aarch64-linux-gnu -mattr=+v8.2a,+i8mm", + topi.arm_cpu.compute_conv2d_NHWC_quantized_interleaved, + topi.arm_cpu.schedule_conv2d_NHWC_quantized_interleaved, + ), ]