Skip to content

Commit

Permalink
Add missing ops for RNNT predictor
Browse files Browse the repository at this point in the history
Summary: As titled. Permute and quantized_linear were not registered properly.

Differential Revision: D56305088
  • Loading branch information
mcremon-meta authored and facebook-github-bot committed Apr 18, 2024
1 parent e69a662 commit 0977dff
Show file tree
Hide file tree
Showing 2 changed files with 11 additions and 0 deletions.
1 change: 1 addition & 0 deletions examples/cadence/ops/CMakeLists.txt
Original file line number Diff line number Diff line change
Expand Up @@ -29,6 +29,7 @@ set(_aten_ops__srcs
"${CMAKE_CURRENT_SOURCE_DIR}/op_embedding.cpp"
"${CMAKE_CURRENT_SOURCE_DIR}/op_full.cpp"
"${CMAKE_CURRENT_SOURCE_DIR}/op_view_copy.cpp"
"${EXECUTORCH_ROOT}/kernels/portable/cpu/op_permute_copy.cpp"
"${EXECUTORCH_ROOT}/kernels/portable/cpu/util/broadcast_util.cpp"
"${EXECUTORCH_ROOT}/kernels/portable/cpu/util/repeat_util.cpp")
add_library(aten_ops_cadence ${_aten_ops__srcs})
Expand Down
10 changes: 10 additions & 0 deletions examples/cadence/ops/functions.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -27,6 +27,11 @@
- arg_meta: null
kernel_name: torch::executor::full_out

- op: permute_copy.out
kernels:
- arg_meta: null
kernel_name: torch::executor::permute_copy_out

- op: view_copy.out
kernels:
- arg_meta: null
Expand All @@ -50,6 +55,11 @@
- arg_meta: null
kernel_name: impl::HiFi::quantized_conv_out

- 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!)
kernels:
- arg_meta: null
kernel_name: impl::HiFi::quantized_layer_norm_out

- 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!)
kernels:
- arg_meta: null
Expand Down

0 comments on commit 0977dff

Please sign in to comment.