【compilation opt】change_cc_test_old_ (#59988)

Galaxy1458 · web-flow · commit 5e66637133dd · 2023-12-15T14:14:26.000+08:00
* change_cc_test_old_

* update

* fix

* update
diff --git a/paddle/fluid/memory/allocation/aligned_allocator.cc b/paddle/fluid/memory/allocation/aligned_allocator.cc
@@ -16,6 +16,9 @@
 
 #include "paddle/fluid/platform/enforce.h"
 
+#include "paddle/fluid/platform/init_phi.h"
+
+REGISTER_FILE_SYMBOLS(aligned_allocator);
 namespace paddle {
 namespace memory {
 namespace allocation {
diff --git a/paddle/fluid/memory/allocation/best_fit_allocator.cc b/paddle/fluid/memory/allocation/best_fit_allocator.cc
@@ -18,6 +18,8 @@
 #include <mutex>
 
 #include "paddle/fluid/platform/enforce.h"
+#include "paddle/fluid/platform/init_phi.h"
+REGISTER_FILE_SYMBOLS(best_fit_allocator);
 
 namespace paddle {
 namespace memory {
diff --git a/paddle/fluid/memory/allocation/buffered_allocator.cc b/paddle/fluid/memory/allocation/buffered_allocator.cc
@@ -13,7 +13,8 @@
 // limitations under the License.
 
 #include "paddle/fluid/memory/allocation/buffered_allocator.h"
-
+#include "paddle/fluid/platform/init_phi.h"
+REGISTER_FILE_SYMBOLS(buffered_allocator);
 namespace paddle {
 namespace memory {
 namespace allocation {
diff --git a/paddle/fluid/pybind/pybind.cc b/paddle/fluid/pybind/pybind.cc
@@ -223,8 +223,13 @@ PYBIND11_MAKE_OPAQUE(paddle::framework::FetchType);
 
 DECLARE_FILE_SYMBOLS(init_phi);
 DECLARE_FILE_SYMBOLS(kernel_dialect);
+DECLARE_FILE_SYMBOLS(buffered_allocator);
+DECLARE_FILE_SYMBOLS(best_fit_allocator);
+DECLARE_FILE_SYMBOLS(aligned_allocator);
+DECLARE_FILE_SYMBOLS(pass_timing);
 DECLARE_FILE_SYMBOLS(op_compatible_info);
 DECLARE_FILE_SYMBOLS(gather_op_handle);
+
 namespace paddle {
 namespace pybind {
 
diff --git a/paddle/pir/pass/pass_timing.cc b/paddle/pir/pass/pass_timing.cc
@@ -18,12 +18,15 @@
 #include <string>
 #include <unordered_map>
 
+#include "paddle/fluid/platform/init_phi.h"
 #include "paddle/pir/core/operation.h"
 #include "paddle/pir/pass/pass.h"
 #include "paddle/pir/pass/pass_instrumentation.h"
 #include "paddle/pir/pass/pass_manager.h"
 #include "paddle/pir/pass/utils.h"
 
+REGISTER_FILE_SYMBOLS(pass_timing);
+
 namespace pir {
 namespace {
 class Timer {
diff --git a/test/cpp/fluid/memory/CMakeLists.txt b/test/cpp/fluid/memory/CMakeLists.txt
@@ -11,8 +11,10 @@ cc_test(
   naive_best_fit_allocator_test
   SRCS naive_best_fit_allocator_test.cc
   DEPS allocator)
-cc_test_old(buffered_allocator_test SRCS buffered_allocator_test.cc DEPS
-            allocator)
+cc_test(
+  buffered_allocator_test
+  SRCS buffered_allocator_test.cc
+  DEPS allocator)
 
 if(WITH_GPU)
   nv_test(
@@ -38,14 +40,21 @@ elseif(WITH_ROCM)
     SRCS best_fit_allocator_test.cc best_fit_allocator_test.cu
     DEPS allocator)
 else()
-  cc_test_old(best_fit_allocator_test SRCS best_fit_allocator_test.cc DEPS
-              allocator)
+  cc_test(
+    best_fit_allocator_test
+    SRCS best_fit_allocator_test.cc
+    DEPS allocator)
 endif()
 
-cc_test_old(test_aligned_allocator SRCS test_aligned_allocator.cc DEPS
-            allocator)
+cc_test(
+  test_aligned_allocator
+  SRCS test_aligned_allocator.cc
+  DEPS allocator)
 
-cc_test_old(retry_allocator_test SRCS retry_allocator_test.cc DEPS allocator)
+cc_test(
+  retry_allocator_test
+  SRCS retry_allocator_test.cc
+  DEPS allocator)
 if(TEST retry_allocator_test)
   set_tests_properties(retry_allocator_test PROPERTIES LABELS
                                                        "RUN_TYPE=EXCLUSIVE")
@@ -118,8 +127,10 @@ cc_test(
   auto_growth_best_fit_allocator_facade_test
   SRCS auto_growth_best_fit_allocator_facade_test.cc
   DEPS allocator)
-cc_test_old(auto_growth_best_fit_allocator_test SRCS
-            auto_growth_best_fit_allocator_test.cc DEPS allocator)
+cc_test(
+  auto_growth_best_fit_allocator_test
+  SRCS auto_growth_best_fit_allocator_test.cc
+  DEPS allocator)
 
 if(NOT WIN32)
   cc_test(
diff --git a/test/cpp/fluid/memory/auto_growth_best_fit_allocator_test.cc b/test/cpp/fluid/memory/auto_growth_best_fit_allocator_test.cc
@@ -99,7 +99,7 @@ class LimitedResourceAllocator : public Allocator {
   const size_t capacity_;
 };
 
-static void TestFreeWhenNoCacheHit(bool free_when_no_cache_hit) {
+void TestFreeWhenNoCacheHit(bool free_when_no_cache_hit) {
   FLAGS_free_idle_chunk = false;
   FLAGS_free_when_no_cache_hit = free_when_no_cache_hit;
   size_t alignment = 256;
