Skip to content

Commit 1c0407a

Browse files
committed
Update
[ghstack-poisoned]
1 parent 923f43e commit 1c0407a

File tree

1 file changed

+9
-0
lines changed

1 file changed

+9
-0
lines changed

shim_et/xplat/executorch/kernels/portable/op_registration_util.bzl

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -246,6 +246,7 @@ ATEN_OPS = (
246246
deps = [
247247
"//executorch/runtime/core/exec_aten/util:scalar_type_util",
248248
"//executorch/runtime/core/exec_aten/util:tensor_util",
249+
"//executorch/kernels/portable/cpu/util:math_util",
249250
"//executorch/kernels/portable/cpu/util:reduce_util",
250251
],
251252
),
@@ -255,6 +256,7 @@ ATEN_OPS = (
255256
"//executorch/runtime/core/exec_aten/util:scalar_type_util",
256257
"//executorch/runtime/core/exec_aten/util:tensor_util",
257258
"//executorch/kernels/portable/cpu/util:index_util",
259+
"//executorch/kernels/portable/cpu/util:math_util",
258260
"//executorch/kernels/portable/cpu/util:reduce_util",
259261
],
260262
),
@@ -278,12 +280,14 @@ ATEN_OPS = (
278280
op_target(
279281
name = "op_argmax",
280282
deps = [
283+
"//executorch/kernels/portable/cpu/util:math_util",
281284
"//executorch/kernels/portable/cpu/util:reduce_util",
282285
],
283286
),
284287
op_target(
285288
name = "op_argmin",
286289
deps = [
290+
"//executorch/kernels/portable/cpu/util:math_util",
287291
"//executorch/kernels/portable/cpu/util:reduce_util",
288292
],
289293
),
@@ -806,6 +810,7 @@ ATEN_OPS = (
806810
op_target(
807811
name = "op_max",
808812
deps = [
813+
"//executorch/kernels/portable/cpu/util:math_util",
809814
"//executorch/kernels/portable/cpu/util:reduce_util",
810815
],
811816
),
@@ -843,6 +848,7 @@ ATEN_OPS = (
843848
op_target(
844849
name = "op_min",
845850
deps = [
851+
"//executorch/kernels/portable/cpu/util:math_util",
846852
"//executorch/kernels/portable/cpu/util:reduce_util",
847853
],
848854
),
@@ -1236,6 +1242,9 @@ ATEN_OPS = (
12361242
),
12371243
op_target(
12381244
name = "op_topk",
1245+
deps = [
1246+
"//executorch/kernels/portable/cpu/util:math_util",
1247+
]
12391248
),
12401249
op_target(
12411250
name = "op_transpose_copy",

0 commit comments

Comments
 (0)