diff --git a/backends/cadence/aot/functions.yaml b/backends/cadence/aot/functions.yaml index 5d85440d90f..9848e2fb9a7 100644 --- a/backends/cadence/aot/functions.yaml +++ b/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 diff --git a/backends/cadence/aot/functions_hifi.yaml b/backends/cadence/aot/functions_hifi.yaml index 9763ecf3ec2..e72a0463178 100644 --- a/backends/cadence/aot/functions_hifi.yaml +++ b/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 diff --git a/backends/cadence/generic/operators/CMakeLists.txt b/backends/cadence/generic/operators/CMakeLists.txt index 77d0b4949a3..dbf06e56bee 100644 --- a/backends/cadence/generic/operators/CMakeLists.txt +++ b/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" diff --git a/backends/cadence/hifi/operators/CMakeLists.txt b/backends/cadence/hifi/operators/CMakeLists.txt index 0e0b18e6e8e..2e764541319 100644 --- a/backends/cadence/hifi/operators/CMakeLists.txt +++ b/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" diff --git a/examples/cadence/operators/test_add_op.py b/examples/cadence/operators/test_add_op.py index 8089fe3a6b1..ab98e674558 100644 --- a/examples/cadence/operators/test_add_op.py +++ b/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__":