add vjp test for new ir

Author: Charles-hit

paddle/fluid/ir/dialect/ir_api.cc

@@ -11,7 +11,6 @@
 // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 // See the License for the specific language governing permissions and
 // limitations under the License.
-#pragma once
 
 #include "paddle/fluid/ir/dialect/ir_api.h"
 #include "paddle/fluid/ir/dialect/pd_dialect.h"

test/cpp/prim/CMakeLists.txt

@@ -61,3 +61,12 @@ if(NOT (NOT WITH_PYTHON AND ON_INFER))
     init_env_utils
     python)
 endif()
+
+# skip win32 since wget is not installed by default on windows machine.
+
+if(NOT WIN32)
+  cc_test(
+    test_vjp_new_ir
+    SRCS test_vjp.cc
+    DEPS phi_kernel_adaptor pd_dialect ir)
+endif()

test/cpp/prim/test_vjp.cc

@@ -0,0 +1,63 @@
+// Copyright (c) 2023 PaddlePaddle Authors. All Rights Reserved.
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//     http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+#include <gtest/gtest.h>
+
+#include "paddle/fluid/framework/new_executor/standalone_executor.h"
+#include "paddle/fluid/ir/dialect/pd_dialect.h"
+#include "paddle/fluid/ir/dialect/pd_op.h"
+#include "paddle/fluid/ir/dialect/pd_type.h"
+#include "paddle/fluid/ir/dialect/utils.h"
+#include "paddle/fluid/ir/interface/op_yaml_info.h"
+#include "paddle/fluid/platform/init_phi.h"
+#include "paddle/ir/core/block.h"
+#include "paddle/ir/core/builtin_attribute.h"
+#include "paddle/ir/core/builtin_dialect.h"
+#include "paddle/ir/core/builtin_op.h"
+#include "paddle/ir/core/ir_context.h"
+#include "paddle/ir/core/program.h"
+#include "paddle/ir/core/utils.h"
+#include "paddle/phi/core/meta_tensor.h"
+#include "paddle/phi/infermeta/binary.h"
+
+DECLARE_FILE_SYMBOLS(kernel_dialect);
+
+PD_DECLARE_KERNEL(full, CPU, ALL_LAYOUT);
+PD_DECLARE_KERNEL(tanh, CPU, ALL_LAYOUT);
+PD_DECLARE_KERNEL(tanh_grad, CPU, ALL_LAYOUT);
+
+TEST(VJP, TanhBackwardTest) {
+  ir::IrContext* ctx = ir::IrContext::Instance();
+  ir::Program program((ctx));
+  ctx->GetOrRegisterDialect<paddle::dialect::PaddleDialect>();
+
+  ir::Builder builder = ir::Builder(ctx, program.block());
+
+  paddle::dialect::FullOp op1 = builder.Build<paddle::dialect::FullOp>(
+      std::vector<int64_t>{1}, 1.0, phi::DataType::FLOAT32, phi::CPUPlace());
+
+  paddle::dialect::TanhOp op2 =
+      builder.Build<paddle::dialect::TanhOp>(op1.out());
+
+  paddle::dialect::FullOp op3 = builder.Build<paddle::dialect::FullOp>(
+      std::vector<int64_t>{1}, 2.0, phi::DataType::FLOAT32, phi::CPUPlace());
+
+  paddle::dialect::VjpInterface tanh_vjp_interface =
+      op2->dyn_cast<paddle::dialect::VjpInterface>();
+
+  std::vector<int> stop_gradients{0};
+  std::vector<ir::OpResult> out_grads{op3.out()};
+  std::vector<ir::OpResult> grad_res =
+      tanh_vjp_interface.Vjp(op2.operation(), out_grads, stop_gradients);
+}