【paddle_test No.17】replace cc_test with paddle_test

paddle/fluid/memory/allocation/allocator.h

@@ -195,7 +195,7 @@ class Allocator : public phi::Allocator {
 
  protected:
   virtual phi::Allocation* AllocateImpl(size_t size) = 0;
-  virtual void FreeImpl(phi::Allocation* allocation);
+  TEST_API virtual void FreeImpl(phi::Allocation* allocation);
   virtual uint64_t ReleaseImpl(const platform::Place& place UNUSED) {
     return 0;
   }

paddle/fluid/memory/allocation/best_fit_allocator.h

@@ -105,7 +105,7 @@ class BestFitAllocation : public Allocation {
 // the prev-chunk and the next-chunk when possible.
 class BestFitAllocator : public Allocator {
  public:
-  explicit BestFitAllocator(phi::Allocation* allocation);
+  TEST_API explicit BestFitAllocator(phi::Allocation* allocation);
 
   void* BasePtr() const { return allocation_->ptr(); }
 

paddle/fluid/memory/allocation/buffered_allocator.h

@@ -32,7 +32,7 @@ namespace allocation {
 // underlying_allocator_
 class BufferedAllocator : public Allocator {
  public:
-  explicit BufferedAllocator(std::shared_ptr<Allocator> allocator);
+  TEST_API explicit BufferedAllocator(std::shared_ptr<Allocator> allocator);
 
   ~BufferedAllocator();
 
@@ -42,7 +42,7 @@ class BufferedAllocator : public Allocator {
   inline void ClearCache() { FreeCache(-1UL); }
 
  private:
-  void FreeCache(size_t size);
+  TEST_API void FreeCache(size_t size);
 
  protected:
   void FreeImpl(phi::Allocation *allocation) override;

paddle/fluid/memory/allocation/cpu_allocator.h

@@ -34,11 +34,11 @@ namespace allocation {
 class CPUAllocator : public Allocator {
  public:
   constexpr static size_t kAlignment = 4096UL;
-  bool IsAllocThreadSafe() const override;
+  TEST_API bool IsAllocThreadSafe() const override;
 
  protected:
   void FreeImpl(phi::Allocation* allocation) override;
-  phi::Allocation* AllocateImpl(size_t size) override;
+  TEST_API phi::Allocation* AllocateImpl(size_t size) override;
 };
 }  // namespace allocation
 }  // namespace memory

paddle/fluid/memory/allocation/naive_best_fit_allocator.h

@@ -35,9 +35,9 @@ class NaiveBestFitAllocator : public Allocator {
   bool IsAllocThreadSafe() const override { return true; }
 
  protected:
-  phi::Allocation *AllocateImpl(size_t size) override;
-  void FreeImpl(phi::Allocation *allocation) override;
-  uint64_t ReleaseImpl(const platform::Place &place) override;
+  TEST_API phi::Allocation *AllocateImpl(size_t size) override;
+  TEST_API void FreeImpl(phi::Allocation *allocation) override;
+  TEST_API uint64_t ReleaseImpl(const platform::Place &place) override;
 
  private:
   platform::Place place_;

paddle/fluid/memory/stats.h

@@ -121,17 +121,21 @@ class Stat : public StatBase {
 // performance than the macro function xxx_MEMORY_STAT_CURRENT_VALUE,
 // xxx_MEMORY_STAT_PEAK_VALUE, and xxx_MEMORY_STAT_UPDATE. Try to use the macro
 // functions where ultra-low performance overhead is required.
-int64_t DeviceMemoryStatCurrentValue(const std::string& stat_type, int dev_id);
-int64_t DeviceMemoryStatPeakValue(const std::string& stat_type, int dev_id);
-void DeviceMemoryStatUpdate(const std::string& stat_type,
-                            int dev_id,
-                            int64_t increment);
-
-int64_t HostMemoryStatCurrentValue(const std::string& stat_type, int dev_id);
-int64_t HostMemoryStatPeakValue(const std::string& stat_type, int dev_id);
-void HostMemoryStatUpdate(const std::string& stat_type,
-                          int dev_id,
-                          int64_t increment);
+TEST_API int64_t DeviceMemoryStatCurrentValue(const std::string& stat_type,
+                                              int dev_id);
+TEST_API int64_t DeviceMemoryStatPeakValue(const std::string& stat_type,
+                                           int dev_id);
+TEST_API void DeviceMemoryStatUpdate(const std::string& stat_type,
+                                     int dev_id,
+                                     int64_t increment);
+
+TEST_API int64_t HostMemoryStatCurrentValue(const std::string& stat_type,
+                                            int dev_id);
+TEST_API int64_t HostMemoryStatPeakValue(const std::string& stat_type,
+                                         int dev_id);
+TEST_API void HostMemoryStatUpdate(const std::string& stat_type,
+                                   int dev_id,
+                                   int64_t increment);
 
 void LogDeviceMemoryStats(const platform::Place& place,
                           const std::string& op_name);

test/cpp/fluid/memory/CMakeLists.txt

@@ -1,17 +1,9 @@
-cc_test(
-  memory_stats_test
-  SRCS memory_stats_test.cc
-  DEPS)
-cc_test(
-  stats_test
-  SRCS stats_test.cc
-  DEPS)
+paddle_test(memory_stats_test SRCS memory_stats_test.cc)
+paddle_test(stats_test SRCS stats_test.cc)
 
-cc_test(
-  naive_best_fit_allocator_test
-  SRCS naive_best_fit_allocator_test.cc
-  DEPS allocator)
-cc_test(
+paddle_test(naive_best_fit_allocator_test SRCS naive_best_fit_allocator_test.cc
+            DEPS allocator)
+nv_test(
   buffered_allocator_test
   SRCS buffered_allocator_test.cc
   DEPS allocator)
@@ -40,35 +32,25 @@ elseif(WITH_ROCM)
     SRCS best_fit_allocator_test.cc best_fit_allocator_test.cu
     DEPS allocator)
 else()
-  cc_test(
-    best_fit_allocator_test
-    SRCS best_fit_allocator_test.cc
-    DEPS allocator)
+  paddle_test(best_fit_allocator_test SRCS best_fit_allocator_test.cc)
 endif()
 
-cc_test(
+nv_test(
   test_aligned_allocator
   SRCS test_aligned_allocator.cc
   DEPS allocator)
 
-cc_test(
-  retry_allocator_test
-  SRCS retry_allocator_test.cc
-  DEPS allocator)
+paddle_test(retry_allocator_test SRCS retry_allocator_test.cc)
 if(TEST retry_allocator_test)
   set_tests_properties(retry_allocator_test PROPERTIES LABELS
                                                        "RUN_TYPE=EXCLUSIVE")
 endif()
 
-cc_test(
-  allocator_facade_abs_flags_test
-  SRCS allocator_facade_abs_flags_test.cc
-  DEPS allocator)
+paddle_test(allocator_facade_abs_flags_test SRCS
+            allocator_facade_abs_flags_test.cc)
 
-cc_test(
-  allocator_facade_frac_flags_test
-  SRCS allocator_facade_frac_flags_test.cc
-  DEPS allocator)
+paddle_test(allocator_facade_frac_flags_test SRCS
+            allocator_facade_frac_flags_test.cc)
 
 if(WITH_GPU)
   nv_test(
@@ -156,3 +138,11 @@ cc_test(
   buddy_allocator_test
   SRCS buddy_allocator_test.cc
   DEPS allocator)
+
+if(WITH_ONNXRUNTIME AND WIN32)
+  # Copy onnxruntime for some c++ test in Windows, since the test will
+  # be build only in CI, so suppose the generator in Windows is Ninja.
+  copy_onnx(memory_stats_test)
+  copy_onnx(stats_test)
+  copy_onnx(naive_best_fit_allocator_test)
+endif()