【paddle_test No.41】replace of cc_test with paddle_test

paddle/fluid/framework/ir/fusion_group/CMakeLists.txt

@@ -2,21 +2,8 @@ cc_library(
   code_generator
   SRCS operation.cc code_generator.cc code_generator_helper.cc
   DEPS graph subgraph_detector)
-if(WITH_GPU OR WITH_ROCM)
-  cc_test(
-    test_code_generator
-    SRCS code_generator_tester.cc
-    DEPS code_generator phi common lod_tensor graph_viz_pass)
-endif()
 
 cc_library(
   fusion_group_pass
   SRCS fusion_group_pass.cc elementwise_group_detector.cc
   DEPS subgraph_detector fuse_pass_base code_generator phi common)
-cc_test(
-  test_fusion_group_pass
-  SRCS fusion_group_pass_tester.cc
-  DEPS fusion_group_pass graph_viz_pass)
-if(WITH_TESTING AND TEST test_code_generator)
-  set_tests_properties(test_code_generator PROPERTIES TIMEOUT 120)
-endif()

paddle/fluid/framework/ir/fusion_group/code_generator.h

@@ -32,14 +32,16 @@ class SubGraph;
 
 class CodeGenerator {
  public:
-  CodeGenerator();
+  TEST_API CodeGenerator();
 
-  std::string Generate(std::string func_name,
-                       const std::vector<OperationExpression>& expressions);
+  TEST_API std::string Generate(
+      std::string func_name,
+      const std::vector<OperationExpression>& expressions);
 
-  std::string Generate(SubGraph* subgraph);
+  TEST_API std::string Generate(SubGraph* subgraph);
 
-  std::vector<OperationExpression> ConvertToExpressions(SubGraph* subgraph);
+  TEST_API std::vector<OperationExpression> ConvertToExpressions(
+      SubGraph* subgraph);
 
  private:
   std::set<int> DistilInputIds(

paddle/fluid/framework/ir/fusion_group/operation.h

@@ -72,7 +72,7 @@ struct Operation {
 
 class OperationMap {
  public:
-  OperationMap();
+  TEST_API OperationMap();
 
   static OperationMap& Instance() {
     PADDLE_ENFORCE_NOT_NULL(
@@ -119,7 +119,7 @@ class OperationMap {
   void InsertMultivariateElementwiseOperations();
 
  private:
-  static OperationMap* map;
+  TEST_API static OperationMap* map;
   std::unordered_map<std::string, Operation> operations_;
   DISABLE_COPY_AND_ASSIGN(OperationMap);
 };

paddle/fluid/framework/ir/node.h

@@ -71,7 +71,7 @@ class Node {
 #if !defined(_WIN32) && (__cplusplus < 201703L)
   static constexpr char kControlDepVarName[] = "__control_var";
 #else
-  static const char kControlDepVarName[];
+  TEST_API static const char kControlDepVarName[];
 #endif
 
   Type NodeType() const { return type_; }

paddle/fluid/framework/ir/pass.h

@@ -356,12 +356,12 @@ struct PassRegistrar : public Registrar {
       &__pass_tmp_registrar_##pass_type##__ UNUSED =        \
           __pass_registrar_##pass_type##__
 
-#define USE_PASS(pass_type)                           \
-  STATIC_ASSERT_PASS_GLOBAL_NAMESPACE(                \
-      __use_pass_itself_##pass_type,                  \
-      "USE_PASS must be called in global namespace"); \
-  extern int TouchPassRegistrar_##pass_type();        \
-  static int use_pass_itself_##pass_type##_ UNUSED =  \
+#define USE_PASS(pass_type)                                 \
+  STATIC_ASSERT_PASS_GLOBAL_NAMESPACE(                      \
+      __use_pass_itself_##pass_type,                        \
+      "USE_PASS must be called in global namespace");       \
+  TEST_API extern int TouchPassRegistrar_##pass_type();     \
+  static int use_pass_itself_##pass_type##_ UNUSED =        \
       TouchPassRegistrar_##pass_type()
 
 }  // namespace ir

paddle/phi/backends/device_code.h

@@ -51,11 +51,11 @@ class DeviceCode {
 #if defined(PADDLE_WITH_CUDA) || defined(PADDLE_WITH_HIP)
 class GPUDeviceCode : public DeviceCode {
  public:
-  explicit GPUDeviceCode(const Place& place,
-                         const std::string& name,
-                         const std::string& kernel);
-  bool Compile(bool include_path = false) override;
-  void Launch(const size_t n, std::vector<void*>* args) const override;
+  TEST_API explicit GPUDeviceCode(const Place& place,
+                                  const std::string& name,
+                                  const std::string& kernel);
+  TEST_API bool Compile(bool include_path = false) override;
+  TEST_API void Launch(const size_t n, std::vector<void*>* args) const override;
 
   void SetNumThreads(int num_threads) { num_threads_ = num_threads; }
   void SetWorkloadPerThread(int workload_per_thread) {

test/cpp/fluid/framework/ir/CMakeLists.txt

@@ -1 +1,2 @@
+add_subdirectory(fusion_group)
 add_subdirectory(memory_optimize_pass)

test/cpp/fluid/framework/ir/fusion_group/CMakeLists.txt

@@ -0,0 +1,13 @@
+if(WITH_GPU OR WITH_ROCM)
+  paddle_test(test_code_generator SRCS code_generator_tester.cc DEPS common)
+endif()
+if(WITH_TESTING AND TEST test_code_generator)
+  set_tests_properties(test_code_generator PROPERTIES TIMEOUT 120)
+endif()
+
+if(NOT WIN32)
+  nv_test(	
+    test_fusion_group_pass	
+    SRCS fusion_group_pass_tester.cc	
+    DEPS fusion_group_pass graph_viz_pass)
+endif()