Fix SVE build: add missing hn::

PiperOrigin-RevId: 632481097
This commit is contained in:
Jan Wassenberg 2024-05-10 06:48:42 -07:00 committed by Copybara-Service
parent c5c9fc300c
commit 22fe9809ac
3 changed files with 21 additions and 7 deletions

View File

@ -10,6 +10,21 @@ package(
],
)
config_setting(
name = "android",
constraint_values = [
"@platforms//os:android",
],
visibility = [":__subpackages__"],
)
FILE_DEPS = select({
"//conditions:default": [
# Placeholder for io deps, do not remove
],
":android": [],
})
cc_library(
name = "io",
srcs = [
@ -18,9 +33,8 @@ cc_library(
],
hdrs = ["io.h"],
deps = [
# Placeholder for io deps, do not remove
"@hwy//:hwy",
],
] + FILE_DEPS,
)
cc_library(

View File

@ -264,10 +264,10 @@ struct CompressTraits<hwy::bfloat16_t> {
i += N;
}
sum0 = Add(sum0, sum1);
sum2 = Add(sum2, sum3);
sum0 = Add(sum0, sum2);
return ReduceSum(df32, sum0);
sum0 = hn::Add(sum0, sum1);
sum2 = hn::Add(sum2, sum3);
sum0 = hn::Add(sum0, sum2);
return hn::ReduceSum(df32, sum0);
}
};

View File

@ -423,7 +423,7 @@ static HWY_INLINE hn::Vec<D> Sigmoid(D d, hn::Vec<D> v) {
VF b3 = hn::Sub(hn::MulAdd(f2, b2, d3), b1);
VF f1 = hn::Sub(hn::MulAdd(f, b3, d4), b2);
return Div(f0, f1);
return hn::Div(f0, f1);
}
// Sigmoid using the logistic function 1 / (1 + exp(-x[i]))