Skip to content

Commit

Permalink
cpu: aarch64: fix wino conv test failure
Browse files Browse the repository at this point in the history
Signed-off-by: Ye Tao <[email protected]>
  • Loading branch information
taoye9 authored and mgouicem committed Sep 30, 2024
1 parent 600230c commit fc621a4
Showing 1 changed file with 4 additions and 2 deletions.
6 changes: 4 additions & 2 deletions tests/gtests/test_iface_wino_convolution.cpp
Original file line number Diff line number Diff line change
@@ -1,6 +1,6 @@
/*******************************************************************************
* Copyright 2020-2023 Intel Corporation
* Copyright 2023 Arm Ltd. and affiliates
* Copyright 2023-2024 Arm Ltd. and affiliates
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
Expand Down Expand Up @@ -68,7 +68,9 @@ TEST_F(wino_conv_test_t, TestSmallPadding) {
if (unsupported_data_type(input.dat_dt)
|| unsupported_data_type(input.wei_dt))
continue;

#if defined(DNNL_AARCH64) && defined(DNNL_AARCH64_USE_ACL)
if (input.dat_dt == data_type::f16) continue;
#endif
memory::desc src_md {{1, 16, 7, 7}, input.dat_dt, tag::any};
memory::desc wei_md {{32, 16, 3, 3}, input.wei_dt, tag::any};
memory::desc dst_md {{1, 32, 7, 7}, input.dat_dt, tag::any};
Expand Down

0 comments on commit fc621a4

Please sign in to comment.