Skip to content

Commit 2804f70

Browse files
committed
Update on "[ExecuTorch] Add broadcast support for optimized add op"
Summary: This brings add op to feature parity, wrt, broadcasting, to mul op in optimized kernels lib Test Plan: tests added Reviewers: Subscribers: Tasks: Tags: cc larryliu0820 manuelcandales Differential Revision: [D69491814](https://our.internmc.facebook.com/intern/diff/D69491814) [ghstack-poisoned]
2 parents 656873f + 8ecbd04 commit 2804f70

File tree

1 file changed

+1
-3
lines changed

1 file changed

+1
-3
lines changed

kernels/optimized/cpu/binary_ops.h

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -199,9 +199,7 @@ Tensor& handle_last_dim_broadcast_elementwise(
199199
const Tensor& a,
200200
const Tensor& b,
201201
Tensor& out,
202-
const ElementwiseOptimizedPath selected_optimized_path,
203-
const executorch::aten::optional<Scalar>& alpha = {}) {
204-
ScalarType out_type = out.scalar_type();
202+
const ElementwiseOptimizedPath selected_optimized_path) {
205203
const Tensor* lhs;
206204
const Tensor* rhs;
207205
if (selected_optimized_path ==

0 commit comments

Comments
 (0)