Skip to content

Commit

Permalink
[FORK][FEATURE][ARM] Enable f16 ACL post-op
Browse files Browse the repository at this point in the history
  • Loading branch information
alvoron authored and luweizhou2016 committed Jul 29, 2024
1 parent c004aad commit e5dbb82
Showing 1 changed file with 2 additions and 4 deletions.
6 changes: 2 additions & 4 deletions src/cpu/acl/acl_post_ops.hpp
Original file line number Diff line number Diff line change
Expand Up @@ -142,10 +142,8 @@ struct acl_post_ops_t {

CHECK(base_post_ops.set_default_formats(&dst_md));
dst_data_type = dst_md.data_type;
// If the first entry is eltwise, we fuse it, except when the datatype
// is fp16 because in this case we want to execute the eltwise in fp32.
if (base_post_ops.len() >= 1 && base_post_ops.entry_[0].is_eltwise()
&& dst_data_type != data_type::f16) {
// If the first entry is eltwise, we fuse it
if (base_post_ops.len() >= 1 && base_post_ops.entry_[0].is_eltwise()) {

const auto &first_po = base_post_ops.entry_[0].eltwise;
ACL_CHECK_SUPPORT(first_po.scale != 1.0f,
Expand Down

0 comments on commit e5dbb82

Please sign in to comment.