Skip to content

Commit

Permalink
cpu: aarch64: fix test test_iface_attr_quantization rls-v3.0
Browse files Browse the repository at this point in the history
  • Loading branch information
davsva01 authored and mgouicem committed Feb 13, 2023
1 parent fbf8dca commit 5bb9070
Showing 1 changed file with 3 additions and 4 deletions.
7 changes: 3 additions & 4 deletions src/cpu/aarch64/jit_uni_binary.cpp
Original file line number Diff line number Diff line change
@@ -1,6 +1,7 @@
/*******************************************************************************
* Copyright 2019-2022 Intel Corporation
* Copyright 2022 FUJITSU LIMITED
* Copyright 2022 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 @@ -258,10 +259,8 @@ bool jit_uni_binary_t::pd_t::alg_preserves_zero() const {
}

bool jit_uni_binary_t::pd_t::check_scales_mask() const {
for (const auto &s : attr()->scales_.scales_) {
if (s.second.mask_ != 0) return false;
}
return true;
const std::vector<int> supported_args = {DNNL_ARG_SRC_0, DNNL_ARG_SRC_1};
return attr_scales_ok(supported_args);
}

bool jit_uni_binary_t::pd_t::is_bcast_pattern(const dims_t &bcast_dims,
Expand Down

0 comments on commit 5bb9070

Please sign in to comment.