diff options
| -rw-r--r-- | BUILD.bazel | 1 | ||||
| -rwxr-xr-x | CMakeLists.txt | 1 | ||||
| -rw-r--r-- | src/init.c | 7 |
3 files changed, 9 insertions, 0 deletions
diff --git a/BUILD.bazel b/BUILD.bazel index fc82eb6aa..5d096cc5f 100644 --- a/BUILD.bazel +++ b/BUILD.bazel @@ -4827,6 +4827,7 @@ PROD_AARCH64_NEONFP16ARITH_MICROKERNEL_SRCS = [ "src/f16-gavgpool/gen/7x-minmax-neonfp16arith-c8.c", "src/f16-gemm/gen/1x16-minmax-neonfp16arith-ld64.c", "src/f16-gemm/gen/6x16-minmax-neonfp16arith-ld64.c", + "src/f16-ibilinear-chw/gen/neonfp16arith-p8.c", "src/f16-ibilinear/gen/neonfp16arith-c8.c", "src/f16-igemm/gen/1x16-minmax-neonfp16arith-ld64.c", "src/f16-igemm/gen/6x16-minmax-neonfp16arith-ld64.c", diff --git a/CMakeLists.txt b/CMakeLists.txt index 14ed78b2e..6cc293739 100755 --- a/CMakeLists.txt +++ b/CMakeLists.txt @@ -3351,6 +3351,7 @@ SET(PROD_AARCH64_NEONFP16ARITH_MICROKERNEL_SRCS src/f16-gavgpool/gen/7x-minmax-neonfp16arith-c8.c src/f16-gemm/gen/1x16-minmax-neonfp16arith-ld64.c src/f16-gemm/gen/6x16-minmax-neonfp16arith-ld64.c + src/f16-ibilinear-chw/gen/neonfp16arith-p8.c src/f16-ibilinear/gen/neonfp16arith-c8.c src/f16-igemm/gen/1x16-minmax-neonfp16arith-ld64.c src/f16-igemm/gen/6x16-minmax-neonfp16arith-ld64.c diff --git a/src/init.c b/src/init.c index 177cf03fc..b1bd94632 100644 --- a/src/init.c +++ b/src/init.c @@ -2824,6 +2824,13 @@ static void init(void) { .ukernel = (xnn_gavgpool_cw_ukernel_function) xnn_f16_gavgpool_cw_ukernel__neonfp16arith_x4, .channel_tile = 4, }; + xnn_params.f32.ibilinear_chw = (struct ibilinear_chw_parameters) { + .ukernel = (xnn_ibilinear_chw_ukernel_function) xnn_f16_ibilinear_chw_ukernel__neonfp16arith_p8, + .channel_tile = 1, + .pixel_tile = 8, + }; + + #endif // XNN_NO_NCHW_OPERATORS #endif // XNN_NO_F16_OPERATORS |