【paddle_test No.1】replace cc_test with paddle_test

paddle/fluid/pir/dialect/op_generator/op_gen.py

@@ -47,6 +47,7 @@
 # Note(Galaxy1458) The need_export_symbol_op_list is used
 # for some unittests these need to export symbol op compiled with dynamic lib.
 need_export_symbol_op_list = [
+    'Add_Op',
     'AbsOp',
     'FullOp',
     'UniformOp',
@@ -55,12 +56,22 @@
     'Conv2dOp',
     'BatchNormOp',
     'FetchOp',
+    'FullIntArrayOp',
     'MatmulOp',
     'SoftmaxOp',
     'ReshapeOp',
     'TransposeOp',
     'LessThanOp',
+    'LayerNormOp',
     'AddGradOp',
+    'ConcatOp',
+    'CummaxOp',
+    'CastOp',
+    'ReluOp',
+    'ReluGradOp',
+    'BatchNorm_Op',
+    'GeluOp',
+    'GeluGradOp',
     'MatmulGradOp',
 ]
 

paddle/fluid/pir/dialect/operator/ir/manual_op.h

@@ -549,10 +549,10 @@ class ExpandOp : public pir::Op<ExpandOp,
                     pir::OperationArgument &argument,  // NOLINT
                     pir::Value x_,                     // NOLINT
                     const std::vector<int64_t> &shape = {});
-  static void Build(pir::Builder &builder,             // NOLINT
-                    pir::OperationArgument &argument,  // NOLINT
-                    pir::Value x_,                     // NOLINT
-                    pir::Value shape_                  // NOLINT
+  TEST_API static void Build(pir::Builder &builder,             // NOLINT
+                             pir::OperationArgument &argument,  // NOLINT
+                             pir::Value x_,                     // NOLINT
+                             pir::Value shape_                  // NOLINT
   );
   static void Build(pir::Builder &builder,             // NOLINT
                     pir::OperationArgument &argument,  // NOLINT

paddle/fluid/pir/drr/include/drr_match_context.h

@@ -17,6 +17,8 @@
 #include <memory>
 #include <string>
 
+#include "paddle/utils/test_macros.h"
+
 namespace pir {
 class Value;
 }
@@ -34,7 +36,7 @@ class MatchContext final {
   const pir::Value& Tensor(const std::string& tensor_name) const;
 
   template <typename T>
-  T Attr(const std::string& attr_name) const;
+  TEST_API T Attr(const std::string& attr_name) const;
 
  private:
   std::shared_ptr<const MatchContextImpl> impl_;

paddle/fluid/pir/drr/include/drr_pattern_context.h

@@ -25,6 +25,7 @@
 #include <vector>
 
 #include "paddle/fluid/pir/drr/include/drr_match_context.h"
+#include "paddle/utils/test_macros.h"
 
 namespace paddle {
 namespace drr {
@@ -105,7 +106,7 @@ class DrrPatternContext {
   DrrPatternContext();
   ~DrrPatternContext() = default;
 
-  drr::SourcePattern SourcePattern();
+  TEST_API drr::SourcePattern SourcePattern();
 
   std::shared_ptr<SourcePatternGraph> source_pattern_graph() const {
     return source_pattern_graph_;
@@ -121,19 +122,20 @@ class DrrPatternContext {
   friend class drr::SourcePattern;
   friend class drr::ResultPattern;
 
-  const Op& SourceOpPattern(
+  TEST_API const Op& SourceOpPattern(
       const std::string& op_type,
       const std::unordered_map<std::string, Attribute>& attributes = {});
-  const drr::Tensor& SourceTensorPattern(const std::string& name);
+  TEST_API const drr::Tensor& SourceTensorPattern(const std::string& name);
 
-  const Op& ResultOpPattern(
+  TEST_API const Op& ResultPattern(
       const std::string& op_type,
       const std::unordered_map<std::string, Attribute>& attributes = {});
-  drr::Tensor& ResultTensorPattern(const std::string& name);
+  TEST_API drr::Tensor& ResultTensorPattern(const std::string& name);
 
   // void RequireEqual(const Attribute& first, const Attribute& second);
   void RequireEqual(const TensorShape& first, const TensorShape& second);
-  void RequireEqual(const TensorDataType& first, const TensorDataType& second);
+  TEST_API void RequireEqual(const TensorDataType& first,
+                             const TensorDataType& second);
   void RequireNativeCall(const ConstraintFunction& custom_fn);
 
   std::shared_ptr<SourcePatternGraph> source_pattern_graph_;
@@ -147,17 +149,17 @@ class Op {
  public:
   const std::string& name() const { return op_type_name_; }
 
-  void operator()(const Tensor& arg, const Tensor* out) const;
+  TEST_API void operator()(const Tensor& arg, const Tensor* out) const;
 
-  Tensor& operator()() const;
+  TEST_API Tensor& operator()() const;
 
-  Tensor& operator()(const Tensor& arg) const;
-  Tensor& operator()(const Tensor& arg0, const Tensor& arg1) const;
+  TEST_API Tensor& operator()(const Tensor& arg) const;
+  TEST_API Tensor& operator()(const Tensor& arg0, const Tensor& arg1) const;
   Tensor& operator()(const Tensor& arg0,
                      const Tensor& arg1,
                      const Tensor& arg2) const;
-  void operator()(const std::vector<const Tensor*>& args,
-                  const std::vector<const Tensor*>& outputs) const;
+  TEST_API void operator()(const std::vector<const Tensor*>& args,
+                           const std::vector<const Tensor*>& outputs) const;
   // const Tensor& operator()(const Tensor& arg0, const Tensor& arg1, const
   // Tensor& arg2) const; const Tensor& operator()(const Tensor& arg0, const
   // Tensor& arg1, const Tensor& arg2, const Tensor& arg3) const; const Tensor&
@@ -201,9 +203,9 @@ class Tensor {
     return name_ == INPUT_NONE_TENSOR_NAME || name_ == OUTPUT_NONE_TENSOR_NAME;
   }
 
-  void Assign(const Tensor& other);
+  TEST_API void Assign(const Tensor& other);
 
-  void operator=(const Tensor& other) const;  // NOLINT
+  TEST_API void operator=(const Tensor& other) const;  // NOLINT
 
   const std::string& name() const { return name_; }
 

paddle/fluid/pir/transforms/transform_general_functions.h

@@ -61,7 +61,7 @@ pir::Type GetDataTypeFromValue(pir::Value value);
  *
  * @return Operation*
  */
-Operation* GetDefiningOpForInput(const Operation* op, uint32_t index);
+TEST_API Operation* GetDefiningOpForInput(const Operation* op, uint32_t index);
 
 /**
  * @brief Get operations and the index of designative op operand (op result)

test/cpp/pir/pattern_rewrite/CMakeLists.txt

@@ -1,26 +1,12 @@
-cc_test(
-  pattern_rewrite_test
-  SRCS pattern_rewrite_test.cc
-  DEPS gtest op_dialect_vjp pir pir_transforms)
+paddle_test(pattern_rewrite_test SRCS pattern_rewrite_test.cc)
 
-cc_test(
-  drr_same_type_binding_test
-  SRCS drr_same_type_binding_test.cc
-  DEPS drr gtest op_dialect_vjp pir pir_transforms)
+paddle_test(drr_same_type_binding_test SRCS drr_same_type_binding_test.cc)
 
-cc_test(
-  drr_fuse_linear_test
-  SRCS drr_fuse_linear_test.cc
-  DEPS pir_transforms drr gtest op_dialect_vjp pir)
+paddle_test(drr_fuse_linear_test SRCS drr_fuse_linear_test.cc)
 
-cc_test(
-  drr_fuse_linear_param_grad_add_test
-  SRCS drr_fuse_linear_param_grad_add_test.cc
-  DEPS pir_transforms drr gtest op_dialect_vjp pir)
+paddle_test(drr_fuse_linear_param_grad_add_test SRCS
+            drr_fuse_linear_param_grad_add_test.cc)
 
 if(WITH_GPU)
-  cc_test(
-    drr_attention_fuse_test
-    SRCS drr_attention_fuse_test.cc
-    DEPS pir_transforms drr gtest op_dialect_vjp pir)
+  paddle_test(drr_attention_fuse_test SRCS drr_attention_fuse_test.cc)
 endif()