From e5362e7e5dbccf81c5296a7e77154e11e1a14d2f Mon Sep 17 00:00:00 2001 From: Pablo Marquez Tello Date: Mon, 16 Oct 2023 14:14:14 +0100 Subject: DirectConv and Im2Col changes to enable fp16 in armv8a multi_isa builds * FP16 kernels must be instantiated in fp16.cpp. * Partially resolves MLCE-1102 Change-Id: Iab9c29dbfd89358f2f663862ff5010c88aeccf8c Signed-off-by: Pablo Marquez Tello Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/10496 Reviewed-by: Anitha Raj Comments-Addressed: Arm Jenkins Tested-by: Arm Jenkins Benchmark: Arm Jenkins --- src/BUILD.bazel | 3 +++ 1 file changed, 3 insertions(+) (limited to 'src/BUILD.bazel') diff --git a/src/BUILD.bazel b/src/BUILD.bazel index 48973dfe0c..a22632e1f5 100644 --- a/src/BUILD.bazel +++ b/src/BUILD.bazel @@ -745,8 +745,11 @@ filegroup( "cpu/kernels/depthwiseconv2d/generic/neon/qasymm8.cpp", "cpu/kernels/depthwiseconv2d/generic/neon/qasymm8_signed.cpp", "cpu/kernels/directconv2d/nchw/all.cpp", + "cpu/kernels/directconv2d/nchw/fp16.cpp", + "cpu/kernels/directconv2d/nhwc/neon/fp16.cpp", "cpu/kernels/directconv2d/nhwc/neon/fp32.cpp", "cpu/kernels/directconv2d/nhwc/neon/impl.cpp", + "cpu/kernels/directconv2d/nhwc/neon/qasymm8.cpp", "cpu/kernels/elementwise_binary/generic/neon/fp16.cpp", "cpu/kernels/elementwise_binary/generic/neon/fp32.cpp", "cpu/kernels/elementwise_binary/generic/neon/integer.cpp", -- cgit v1.2.1