Cadence: Register missing portable ops (neg, exp, native_layer_norm)

Differential Revision: D99870895

Pull Request resolved: pytorch#18745

Author: aliafzal

backends/cadence/aot/functions.yaml

@@ -62,6 +62,11 @@
     - arg_meta: null
       kernel_name: torch::executor::empty_out
 
+- op: exp.out
+  kernels:
+    - arg_meta: null
+      kernel_name: torch::executor::exp_out
+
 - op: expand_copy.out
   kernels:
     - arg_meta: null
@@ -92,6 +97,16 @@
     - arg_meta: null
       kernel_name: torch::executor::mean_dim_out
 
+- op: native_layer_norm.out
+  kernels:
+    - arg_meta: null
+      kernel_name: torch::executor::native_layer_norm_out
+
+- op: neg.out
+  kernels:
+    - arg_meta: null
+      kernel_name: torch::executor::neg_out
+
 - op: mul.out
   kernels:
     - arg_meta: null

backends/cadence/aot/functions_hifi.yaml

@@ -102,6 +102,11 @@
     - arg_meta: null
       kernel_name: impl::HiFi::eq_Tensor_out
 
+- op: exp.out
+  kernels:
+    - arg_meta: null
+      kernel_name: torch::executor::exp_out
+
 - op: fmod.Tensor_out
   kernels:
     - arg_meta: null
@@ -207,6 +212,11 @@
     - arg_meta: null
       kernel_name: impl::HiFi::mul_out
 
+- op: neg.out
+  kernels:
+    - arg_meta: null
+      kernel_name: torch::executor::neg_out
+
 - op: ne.Tensor_out
   kernels:
     - arg_meta: null

backends/cadence/generic/operators/CMakeLists.txt

@@ -33,11 +33,14 @@ set(_aten_ops__srcs
     "${EXECUTORCH_ROOT}/kernels/portable/cpu/op_clone.cpp"
     "${EXECUTORCH_ROOT}/kernels/portable/cpu/op_div.cpp"
     "${EXECUTORCH_ROOT}/kernels/portable/cpu/op_embedding.cpp"
+    "${EXECUTORCH_ROOT}/kernels/portable/cpu/op_exp.cpp"
     "${EXECUTORCH_ROOT}/kernels/portable/cpu/op_full.cpp"
     "${EXECUTORCH_ROOT}/kernels/portable/cpu/op_hardtanh.cpp"
     "${EXECUTORCH_ROOT}/kernels/portable/cpu/op_max_pool2d_with_indices.cpp"
     "${EXECUTORCH_ROOT}/kernels/portable/cpu/op_mean.cpp"
     "${EXECUTORCH_ROOT}/kernels/portable/cpu/op_mul.cpp"
+    "${EXECUTORCH_ROOT}/kernels/portable/cpu/op_native_layer_norm.cpp"
+    "${EXECUTORCH_ROOT}/kernels/portable/cpu/op_neg.cpp"
     "${EXECUTORCH_ROOT}/kernels/portable/cpu/op_permute_copy.cpp"
     "${EXECUTORCH_ROOT}/kernels/portable/cpu/op_rsqrt.cpp"
     "${EXECUTORCH_ROOT}/kernels/portable/cpu/op_sigmoid.cpp"

backends/cadence/hifi/operators/CMakeLists.txt

@@ -57,9 +57,11 @@ set(_aten_ops__srcs
     "${EXECUTORCH_ROOT}/backends/cadence/hifi/operators/op_view_copy.cpp"
     "${EXECUTORCH_ROOT}/backends/cadence/hifi/operators/op_where.cpp"
     "${EXECUTORCH_ROOT}/kernels/portable/cpu/op_clone.cpp"
+    "${EXECUTORCH_ROOT}/kernels/portable/cpu/op_exp.cpp"
     "${EXECUTORCH_ROOT}/kernels/portable/cpu/op_gelu.cpp"
     "${EXECUTORCH_ROOT}/kernels/portable/cpu/op_max_pool2d_with_indices.cpp"
     "${EXECUTORCH_ROOT}/kernels/portable/cpu/op_native_layer_norm.cpp"
+    "${EXECUTORCH_ROOT}/kernels/portable/cpu/op_neg.cpp"
     "${EXECUTORCH_ROOT}/kernels/portable/cpu/util/normalization_ops_util.cpp"
     "${EXECUTORCH_ROOT}/kernels/portable/cpu/op_to_copy.cpp"
     "${EXECUTORCH_ROOT}/kernels/portable/cpu/pattern/unary_ufunc_realhbbf16_to_floathbf16.cpp"

examples/cadence/operators/test_add_op.py

@@ -64,9 +64,7 @@ def forward(self, x: torch.Tensor, y: torch.Tensor):
         Y = torch.randn(Yshape)
 
         model.eval()
-        export_and_run_model(
-            model, (X, Y), file_name=self._testMethodName, run_and_compare=False
-        )
+        export_and_run_model(model, (X, Y), file_name=self._testMethodName)
 
     # pyre-fixme[16]: Module `parameterized.parameterized` has no attribute `expand`.
     @parameterized.expand(
@@ -114,9 +112,7 @@ def forward(self, x: torch.Tensor, y: float):
         Y = 2.34
 
         model.eval()
-        export_and_run_model(
-            model, (X, Y), file_name=self._testMethodName, run_and_compare=False
-        )
+        export_and_run_model(model, (X, Y), file_name=self._testMethodName)
 
 
 if __name__ == "__main__":