phase 1 linkage

Differential Revision: D84639743

Author: Zonglin Peng

backends/cadence/aot/functions_hifi.yaml

@@ -472,7 +472,12 @@
 - func: cadence::im2row.out(Tensor input, int[2] kernel_size, int[2] dilation, int[2] padding, int[2] stride, Tensor in_zero_point, bool channel_last=False, *, Tensor(a!) out) -> Tensor(a!)
   kernels:
     - arg_meta: null
-      kernel_name: cadence::impl::HiFi::im2row_out
+      kernel_name: impl::HiFi::native::im2row_out
+
+- func: cadence::im2row.per_tensor_out(Tensor input, int[2] kernel_size, int[2] dilation, int[2] padding, int[2] stride, int in_zero_point, bool channel_last=False, *, Tensor(a!) out) -> Tensor(a!)
+  kernels:
+    - arg_meta: null
+      kernel_name: impl::HiFi::native::im2row_per_tensor_out
 
 - func: cadence::quantized_linear_asym8sxasym8s_asym8s.per_tensor_out(Tensor src, Tensor weight, Tensor bias, int src_zero_point, int weight_zero_point, int out_multiplier, int out_shift, int out_zero_point, Tensor? offset, *, Tensor(a!) out) -> Tensor(a!)
   kernels:

backends/cadence/hifi/operators/CMakeLists.txt

@@ -16,7 +16,7 @@ include(${EXECUTORCH_ROOT}/tools/cmake/Codegen.cmake)
 
 # ATen compliant ops that are needed to run this model.
 set(_aten_ops__srcs
-    "${EXECUTORCH_ROOT}/backends/cadence/hifi/operators/im2row_out.cpp"
+    "${EXECUTORCH_ROOT}/backends/cadence/hifi/operators/op_im2row_out.cpp"
     "${EXECUTORCH_ROOT}/backends/cadence/hifi/operators/op_add.cpp"
     "${EXECUTORCH_ROOT}/backends/cadence/hifi/operators/op_atan2.cpp"
     "${EXECUTORCH_ROOT}/backends/cadence/hifi/operators/op_bitwise_and.cpp"

backends/cadence/hifi/operators/im2row_out.cpp renamed to backends/cadence/hifi/operators/op_im2row_out.cpp

@@ -19,7 +19,6 @@ using ::executorch::aten::ScalarType;
 using ::executorch::aten::Tensor;
 using ::executorch::runtime::KernelRuntimeContext;
 
-namespace cadence {
 namespace impl {
 namespace HiFi {
 namespace native {
@@ -172,7 +171,7 @@ void im2row_out(
     optimized = true;
 
   if (!optimized) {
-    WORD8* ptr1 = (WORD8*)kernels::allocate_temp_memory(
+    WORD8* ptr1 = (WORD8*)::impl::HiFi::kernels::allocate_temp_memory(
         ctx, ((batch_size * in_c * in_h * in_w) + 8) * sizeof(WORD8));
 
     WORD8* pin = (WORD8*)ALIGN_PTR(ptr1, 8);
@@ -325,7 +324,7 @@ void im2row_per_tensor_out(
     optimized = true;
 
   if (!optimized) {
-    WORD8* ptr1 = (WORD8*)kernels::allocate_temp_memory(
+    WORD8* ptr1 = (WORD8*)::impl::HiFi::kernels::allocate_temp_memory(
         ctx, ((batch_size * in_c * in_h * in_w) + 8) * sizeof(WORD8));
 
     WORD8* pin = (WORD8*)ALIGN_PTR(ptr1, 8);
@@ -428,4 +427,3 @@ void im2row_per_tensor_out(
 } // namespace native
 } // namespace HiFi
 } // namespace impl
-} // namespace cadence

backends/cadence/hifi/operators/targets.bzl

@@ -53,6 +53,7 @@ OPERATORS = [
     "ge",
     "gt",
     "hardtanh",
+    "im2row_out",
     "le",
     "lt",
     "masked_fill",