Skip to content

Commit 6ee91e2

Browse files
Update on "[ET][Portable][Build Size] REALHBF16 binary ops: maximum, minimum, mul"
- mul: 1.69 M -> 15 K - maximum: 353 K -> 11 K - minimum: 353 K -> 11 K Differential Revision: [D63909726](https://our.internmc.facebook.com/intern/diff/D63909726/) [ghstack-poisoned]
2 parents c3085d5 + 52dc137 commit 6ee91e2

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

kernels/portable/cpu/op_mul.cpp

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -45,7 +45,8 @@ Tensor& mul_out(
4545

4646
ET_KERNEL_CHECK(
4747
ctx,
48-
(executorch::runtime::isRealType(compute_type) || compute_type == ScalarType::Bool),
48+
(executorch::runtime::isRealType(compute_type) ||
49+
compute_type == ScalarType::Bool),
4950
InvalidArgument,
5051
out);
5152

@@ -93,7 +94,7 @@ Tensor& mul_scalar_out(
9394
ET_SWITCH_REALB_TYPES(compute_type, ctx, op_name, CTYPE_COMPUTE, [&]() {
9495
const CTYPE_COMPUTE val_b = utils::scalar_to<CTYPE_COMPUTE>(b);
9596
utils::apply_unitensor_elementwise_fn<CTYPE_COMPUTE, op_name>(
96-
[val_b](const CTYPE_COMPUTE val_a) {return val_a * val_b;},
97+
[val_b](const CTYPE_COMPUTE val_a) { return val_a * val_b; },
9798
ctx,
9899
a,
99100
utils::SupportedTensorDtypes::REALHBBF16,

0 commit comments

Comments
 (0)