Skip to content

Commit e9fe6af

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 [ghstack-poisoned]
2 parents 7ea55eb + ffb6903 commit e9fe6af

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

kernels/optimized/cpu/op_mul.cpp

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -133,8 +133,7 @@ Tensor& opt_mul_out(
133133
// Reason for using alpha even when used for mul is becasuse
134134
// handle_broadcast_elementwise is used for add and sub as well
135135
// and it uses alpha.
136-
auto mul_lambda = [](auto x, auto y, auto alpha) {
137-
[[maybe_unused]] alpha;
136+
auto mul_lambda = [](auto x, auto y, [[maybe_unused]] auto alpha) {
138137
return x * y;
139138
};
140139
static constexpr const char op_name[] = "mul.out";

0 commit comments

Comments
 (0)