Skip to content

Commit 8e0e9fd

Browse files
committed
Update base for Update on "[Executorch] Add broadcasting support to optimized op_sub"
Summary: This diff builds on top of previous one to add support for limited handling of broadcasting for sub Test Plan: tests added Reviewers: Subscribers: Tasks: Tags: cc larryliu0820 manuelcandales Differential Revision: [D69491818](https://our.internmc.facebook.com/intern/diff/D69491818) [ghstack-poisoned]
1 parent 012fe19 commit 8e0e9fd

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)