add test for tanh vjp

Author: Charles-hit

paddle/fluid/ir/dialect/vjp_interface.cc

@@ -32,7 +32,7 @@ std::vector<ir::OpResult> TanhOp::Vjp(ir::Operation* op,
   std::vector<ir::OpResult> res;
   res.reserve(tensor_res.size());
   // TODO(wanghao107): maybe combile here
-  for (int i = 0; i < tensor_res.size(); ++i) {
+  for (size_t i = 0; i < tensor_res.size(); ++i) {
     res.emplace_back(
         std::static_pointer_cast<primitive::experimental::DescTensor>(
             tensor_res[i][0].impl())

test/cpp/prim/test_vjp.cc

@@ -14,12 +14,14 @@
 
 #include <gtest/gtest.h>
 
+#include "paddle/fluid/framework/new_executor/new_ir_interpreter.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/ir/transforms/pd_op_to_kernel_pass.h"
 #include "paddle/fluid/platform/init_phi.h"
 #include "paddle/ir/core/block.h"
 #include "paddle/ir/core/builtin_attribute.h"
@@ -28,14 +30,14 @@
 #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);
+namespace paddle {
+namespace framework {
 
 TEST(VJP, TanhBackwardTest) {
   ir::IrContext* ctx = ir::IrContext::Instance();
@@ -60,4 +62,39 @@ TEST(VJP, TanhBackwardTest) {
   std::vector<ir::OpResult> out_grads{op3.out()};
   std::vector<ir::OpResult> grad_res =
       tanh_vjp_interface.Vjp(op2.operation(), out_grads, stop_gradients);
+
+  std::ostringstream print_stream;
+  program.Print(print_stream);
+  std::cout << print_stream.str();
+
+  auto kernel_program = paddle::dialect::PdOpLowerToKernelPass(&program);
+
+  auto place = platform::CPUPlace();
+  Scope scope;
+
+  ProgramDesc prog_desc;
+  InterpreterCore test_core(place, std::move(kernel_program), &scope);
+  test_core.BetaRun({});
+  std::stringstream os;
+  os << reinterpret_cast<NewIRInterpreter*>(
+      const_cast<InterpreterBaseImpl*>(test_core.Impl()));
+  std::string prefix_str = os.str();
+  auto out_tensor =
+      test_core.local_scope() == nullptr
+          ? scope.FindVar(prefix_str + "_inner_var_1")->Get<phi::DenseTensor>()
+          : test_core.local_scope()
+                ->FindVar(prefix_str + "_inner_var_1")
+                ->Get<phi::DenseTensor>();
+  auto grad_out_tensor =
+      test_core.local_scope() == nullptr
+          ? scope.FindVar(prefix_str + "_inner_var_3")->Get<phi::DenseTensor>()
+          : test_core.local_scope()
+                ->FindVar(prefix_str + "_inner_var_3")
+                ->Get<phi::DenseTensor>();
+
+  ASSERT_NEAR(out_tensor.data<float>()[0], 0.76159, 1e-5);
+  ASSERT_NEAR(grad_out_tensor.data<float>()[0], 0.83995, 1e-5);
 }
+
+}  // namespace framework
+}  // namespace paddle