diff --git a/backends/cadence/hifi/operators/op_bmm.cpp b/backends/cadence/hifi/operators/op_bmm.cpp index 31ae966966d..0262703bb73 100644 --- a/backends/cadence/hifi/operators/op_bmm.cpp +++ b/backends/cadence/hifi/operators/op_bmm.cpp @@ -16,8 +16,8 @@ using exec_aten::ScalarType; using executorch::runtime::KernelRuntimeContext; using executorch::runtime::kTensorDimensionLimit; using executorch::runtime::resize_tensor; -using executorch::runtime::tensors_have_same_dim_order; using executorch::runtime::tensor_is_default_dim_order; +using executorch::runtime::tensors_have_same_dim_order; using torch::executor::check_bmm_args; using torch::executor::Error; using torch::executor::get_bmm_out_target_size; @@ -78,16 +78,16 @@ Tensor& bmm_out( WORD32 out_stride = p; WORD32* __restrict__ tmp = - (WORD32* __restrict__)kernels::allocate_temp_memory( - ctx, (batch_size * m * p) * sizeof(float)); + (WORD32* __restrict__)kernels::allocate_temp_memory( + ctx, (batch_size * m * p) * sizeof(float)); ET_KERNEL_CHECK(ctx, tmp != nullptr, MemoryAllocationFailed, out); tmp[batch_size * m * p] = {0}; WORD32* __restrict__ p_o = - (WORD32* __restrict__)kernels::allocate_temp_memory( - ctx, (batch_size * m * p) * sizeof(WORD32)); + (WORD32* __restrict__)kernels::allocate_temp_memory( + ctx, (batch_size * m * p) * sizeof(WORD32)); ET_KERNEL_CHECK(ctx, p_o != nullptr, MemoryAllocationFailed, out); diff --git a/backends/cadence/hifi/operators/op_mm.cpp b/backends/cadence/hifi/operators/op_mm.cpp index ceedc97eeb3..334eba0a158 100644 --- a/backends/cadence/hifi/operators/op_mm.cpp +++ b/backends/cadence/hifi/operators/op_mm.cpp @@ -76,8 +76,8 @@ Tensor& mm_out( WORD32 out_stride = p; WORD32* __restrict__ p_o = - (WORD32* __restrict__)kernels::allocate_temp_memory( - ctx, (n * p) * sizeof(WORD32)); + (WORD32* __restrict__)kernels::allocate_temp_memory( + ctx, (n * p) * sizeof(WORD32)); WORD32 p_inp_shape[2]; p_inp_shape[0] = n; diff --git a/backends/cadence/hifi/operators/targets.bzl b/backends/cadence/hifi/operators/targets.bzl index 61a526b4073..5c91c7a4c74 100644 --- a/backends/cadence/hifi/operators/targets.bzl +++ b/backends/cadence/hifi/operators/targets.bzl @@ -34,6 +34,11 @@ def define_operator(name: str, deps: list[str] | None = None) -> None: OPERATORS = [ "add", "atan2", + "bmm", + "mm", + "slice_copy", + "split_with_sizes_copy", + "view_copy", "cat", "clamp", "dequantize_per_tensor",