Buckify op_add for FusionG3 and add cxx tests.

Differential Revision: D66532566

Author: hsharma35

backends/cadence/fusion_g3/operators/operators.h

@@ -0,0 +1,34 @@
+/*
+ * 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/runtime/core/exec_aten/exec_aten.h>
+#include <executorch/runtime/kernel/kernel_runtime_context.h>
+
+namespace cadence {
+namespace impl {
+namespace G3 {
+namespace native {
+
+::executorch::aten::Tensor& add_out(
+    ::executorch::runtime::KernelRuntimeContext& ctx,
+    const ::executorch::aten::Tensor& a,
+    const ::executorch::aten::Tensor& b,
+    const ::executorch::aten::Scalar& alpha,
+    ::executorch::aten::Tensor& out);
+
+::executorch::aten::Tensor& add_scalar_out(
+    ::executorch::runtime::KernelRuntimeContext& ctx,
+    const ::executorch::aten::Tensor& a,
+    const ::executorch::aten::Scalar& b,
+    const ::executorch::aten::Scalar& alpha,
+    ::executorch::aten::Tensor& out);
+
+} // namespace native
+} // namespace G3
+} // namespace impl
+} // namespace cadence

backends/cadence/fusion_g3/operators/tests/test_op_add.cpp

@@ -0,0 +1,85 @@
+/*
+ * 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 <gtest/gtest.h>
+
+#include <executorch/backends/cadence/fusion_g3/operators/operators.h>
+#include <executorch/runtime/core/error.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 <executorch/runtime/platform/runtime.h>
+
+namespace cadence {
+namespace impl {
+namespace G3 {
+namespace native {
+namespace {
+
+using ::executorch::aten::Scalar;
+using ::executorch::aten::ScalarType;
+using ::executorch::aten::SizesType;
+using ::executorch::aten::StridesType;
+using ::executorch::aten::Tensor;
+using ::executorch::runtime::runtime_init;
+using ::executorch::runtime::testing::TensorFactory;
+using ::testing::Test;
+
+class FusionG3OperatorTest : public Test {
+ public:
+  void SetUp() override {
+    runtime_init();
+  }
+
+ protected:
+  Tensor&
+  add_out(const Tensor& a, const Tensor& b, const Scalar& alpha, Tensor& out) {
+    return cadence::impl::G3::native::add_out(context_, a, b, alpha, out);
+  }
+
+  KernelRuntimeContext context_;
+};
+
+TEST_F(FusionG3OperatorTest, TwoDimFloatTensorAddTest) {
+  TensorFactory<ScalarType::Float> tf;
+  const std::vector<int> sizes{2, 2};
+  Tensor out = tf.zeros(sizes);
+
+  // Add two 2x2 tensors.
+  add_out(tf.make(sizes, {1, 2, 3, 4}), tf.make(sizes, {2, 2, 2, 2}), 1, out);
+
+  EXPECT_TENSOR_EQ(out, tf.make(sizes, {3, 4, 5, 6}));
+}
+
+TEST_F(FusionG3OperatorTest, TensorScalarAddTest) {
+  TensorFactory<ScalarType::Float> tf;
+  const std::vector<int> sizes{2, 2};
+  Tensor out = tf.zeros(sizes);
+
+  // Add 2x2 tensor with scalar.
+  add_out(tf.make(sizes, {1, 2, 3, 4}), tf.make({1}, {2}), 1, out);
+
+  EXPECT_TENSOR_EQ(out, tf.make(sizes, {3, 4, 5, 6}));
+}
+
+TEST_F(FusionG3OperatorTest, AddWithBroadcastTest) {
+  TensorFactory<ScalarType::Float> tf;
+  // Broadcast add with two dimensions -- 1x16x8x8 and 8x8.
+  const std::vector<int> sizeOfA{1, 16, 8, 8}, sizeOfB{8, 8};
+  Tensor out = tf.zeros(sizeOfA);
+
+  add_out(tf.ones(sizeOfA), tf.ones(sizeOfB), 1, out);
+
+  EXPECT_TENSOR_EQ(out, tf.full(sizeOfA, 2));
+}
+
+} // namespace
+} // namespace native
+} // namespace G3
+} // namespace impl
+} // namespace cadence