Add ceil op

Reviewed By: SS-JIA, kirklandsign

Differential Revision: D48380140

fbshipit-source-id: 79da1b45f44dee2fd18f328d4587f469c4f4f898

Author: manuelcandales

kernels/aten/functions.yaml

@@ -81,6 +81,8 @@
 
 - op: cat.out
 
+- op: ceil.out
+
 - op: clamp_min.out
 
 - op: clamp.out

kernels/portable/cpu/op_ceil.cpp

@@ -0,0 +1,25 @@
+/*
+ * Copyright (c) Meta Platforms, Inc. and affiliates.
+ * All rights reserved.
+ *
+ * This source code is licensed under the BSD-style license found in the
+ * LICENSE file in the root directory of this source tree.
+ */
+
+#include <executorch/kernels/portable/cpu/pattern/pattern.h>
+#include <executorch/runtime/kernel/kernel_includes.h>
+#include <cmath>
+
+namespace torch {
+namespace executor {
+namespace native {
+
+using exec_aten::Tensor;
+
+Tensor& ceil_out(RuntimeContext& ctx, const Tensor& in, Tensor& out) {
+  return internal::unary_ufunc_real(std::ceil, ctx, in, out);
+}
+
+} // namespace native
+} // namespace executor
+} // namespace torch

kernels/portable/cpu/targets.bzl

@@ -178,6 +178,12 @@ _ATEN_OPS = (
             "//executorch/kernels/portable/cpu/util:copy_ops_util",
         ],
     ),
+    op_target(
+        name = "op_ceil",
+        deps = [
+            "//executorch/kernels/portable/cpu/pattern:pattern",
+        ],
+    ),
     op_target(
         name = "op_clamp",
         deps = [

kernels/portable/functions.yaml

@@ -177,6 +177,11 @@
     - arg_meta: null
       kernel_name: torch::executor::cat_out
 
+- op: ceil.out
+  kernels:
+    - arg_meta: null
+      kernel_name: torch::executor::ceil_out
+
 - op: clamp.out
   cpp_no_default_args: ['min']
   kernels:

kernels/test/op_ceil_test.cpp

@@ -0,0 +1,38 @@
+/*
+ * Copyright (c) Meta Platforms, Inc. and affiliates.
+ * All rights reserved.
+ *
+ * This source code is licensed under the BSD-style license found in the
+ * LICENSE file in the root directory of this source tree.
+ */
+
+#include <executorch/kernels/test/FunctionHeaderWrapper.h> // Declares the operator
+#include <executorch/kernels/test/TestUtil.h>
+#include <executorch/runtime/core/exec_aten/exec_aten.h>
+#include <executorch/runtime/core/exec_aten/testing_util/tensor_factory.h>
+#include <executorch/runtime/core/exec_aten/testing_util/tensor_util.h>
+
+#include <gtest/gtest.h>
+
+using namespace ::testing;
+using exec_aten::ScalarType;
+using exec_aten::Tensor;
+using torch::executor::testing::TensorFactory;
+
+Tensor& op_ceil_out(const Tensor& self, Tensor& out) {
+  exec_aten::RuntimeContext context{};
+  return torch::executor::aten::ceil_outf(context, self, out);
+}
+
+TEST(OpCeilTest, SanityCheck) {
+  TensorFactory<ScalarType::Float> tf;
+
+  Tensor in = tf.make({1, 7}, {-3.0, -2.99, -1.01, 0.0, 1.01, 2.99, 3.0});
+  Tensor out = tf.zeros({1, 7});
+  Tensor expected = tf.make({1, 7}, {-3.0, -2.0, -1.0, 0.0, 2.0, 3.0, 3.0});
+
+  Tensor ret = op_ceil_out(in, out);
+
+  EXPECT_TENSOR_EQ(out, ret);
+  EXPECT_TENSOR_EQ(out, expected);
+}

kernels/test/targets.bzl

@@ -179,6 +179,7 @@ def define_common_targets():
     _common_op_test("op_bitwise_xor_test", ["aten", "portable"])
     _common_op_test("op_bmm_test", ["aten", "portable", "optimized"])
     _common_op_test("op_cat_test", ["aten", "portable"])
+    _common_op_test("op_ceil_test", ["aten", "portable"])
     _common_op_test("op_clamp_test", ["aten", "portable"])
     _common_op_test("op_clone_test", ["aten", "portable"])
     _common_op_test("op_constant_pad_nd_test", ["aten", "portable"])