Skip to content

Commit 09984ac

Browse files
mcremon-metafacebook-github-bot
authored andcommitted
Add missing ops for RNNT predictor (#3125)
Summary: As titled. Permute and quantized_linear were not registered properly. Differential Revision: D56305088
1 parent e69a662 commit 09984ac

File tree

2 files changed

+12
-0
lines changed

2 files changed

+12
-0
lines changed

examples/cadence/ops/CMakeLists.txt

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -29,6 +29,8 @@ set(_aten_ops__srcs
2929
"${CMAKE_CURRENT_SOURCE_DIR}/op_embedding.cpp"
3030
"${CMAKE_CURRENT_SOURCE_DIR}/op_full.cpp"
3131
"${CMAKE_CURRENT_SOURCE_DIR}/op_view_copy.cpp"
32+
"${EXECUTORCH_ROOT}/kernels/portable/cpu/op_permute_copy.cpp"
33+
"${EXECUTORCH_ROOT}/kernels/portable/cpu/util/copy_ops_util.cpp"
3234
"${EXECUTORCH_ROOT}/kernels/portable/cpu/util/broadcast_util.cpp"
3335
"${EXECUTORCH_ROOT}/kernels/portable/cpu/util/repeat_util.cpp")
3436
add_library(aten_ops_cadence ${_aten_ops__srcs})

examples/cadence/ops/functions.yaml

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -27,6 +27,11 @@
2727
- arg_meta: null
2828
kernel_name: torch::executor::full_out
2929

30+
- op: permute_copy.out
31+
kernels:
32+
- arg_meta: null
33+
kernel_name: torch::executor::permute_copy_out
34+
3035
- op: view_copy.out
3136
kernels:
3237
- arg_meta: null
@@ -50,6 +55,11 @@
5055
- arg_meta: null
5156
kernel_name: impl::HiFi::quantized_conv_out
5257

58+
- func: cadence::quantized_layer_norm.out(Tensor input, Tensor in_scale, Tensor in_zero_point, int[] normalized_shape, Tensor weight, Tensor bias, float eps, float output_scale, int output_zero_point, *, Tensor(a!) out) -> Tensor(a!)
59+
kernels:
60+
- arg_meta: null
61+
kernel_name: impl::HiFi::quantized_layer_norm_out
62+
5363
- func: cadence::quantized_linear.out(Tensor src, Tensor weight, Tensor bias, float src_scale, int src_zero_point, float weight_scale, int weight_zero_point, Tensor out_multiplier, Tensor out_shift, int out_zero_point, *, Tensor(a!) out) -> Tensor(a!)
5464
kernels:
5565
- arg_meta: null

0 commit comments

Comments
 (0)