fix ci

PaddlePaddle · liuruyan · Nov 14, 2025 · Nov 11, 2025 · Nov 11, 2025 · Nov 11, 2025
commit e15aea08e6a190ebf659873978047c450391c1bd
diff --git a/test/cpp/fluid/memory/CMakeLists.txt b/test/cpp/fluid/memory/CMakeLists.txt
@@ -6,6 +6,10 @@ cc_test(
   stats_test
   SRCS stats_test.cc
   DEPS)
+cc_test(
+  compact_allocator_test
+  SRCS compact_allocator_test.cc
+  DEPS)
 
 cc_test(
   naive_best_fit_allocator_test

diff --git a/test/cpp/fluid/memory/compact_allocator_test.cc b/test/cpp/fluid/memory/compact_allocator_test.cc
@@ -0,0 +1,47 @@
+// Copyright (c) 2018 PaddlePaddle Authors. All Rights Reserved.
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//     http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+#include "paddle/phi/core/memory/allocation/allocator.h"
+#include "paddle/phi/core/memory/allocation/retry_allocator.h"
+
+#include <thread>  // NOLINT
+
+#include "gtest/gtest.h"
+#include "paddle/phi/core/memory/allocation/best_fit_allocator.h"
+#include "paddle/phi/core/memory/allocation/cpu_allocator.h"
+#if defined(PADDLE_WITH_CUDA) || defined(PADDLE_WITH_HIP)
+#include "paddle/phi/core/memory/allocation/cuda_allocator.h"
+#endif
+namespace paddle {
+namespace memory {
+namespace allocation {
+
+TEST(RetryAllocator, TestRetryComopact) {
+  CPUAllocator cpu_allocator;
+
+  size_t size = (1 << 20);
+  auto cpu_allocation = cpu_allocator.Allocate(size);
+
+  // Reserve to perform more tests in the future
+  std::unique_ptr<BestFitAllocator> best_fit_allocator(
+      new BestFitAllocator(cpu_allocation.get()));
+  std::shared_ptr<Allocator> allocator = std::make_shared<RetryAllocator>(
+      std::move(best_fit_allocator), phi::CPUPlace(), 1);
+
+  allocator->Compact(phi::CPUPlace());
+}
+
+}  // namespace allocation
+}  // namespace memory
+}  // namespace paddle
diff --git a/test/legacy_test/test_vmm_compact.py b/test/legacy_test/test_vmm_compact.py
@@ -20,7 +20,8 @@
 
 
 @unittest.skipIf(
-    not paddle.is_compiled_with_cuda(), 'should compile with cuda.'
+    (not paddle.is_compiled_with_cuda()) or paddle.is_compiled_with_rocm,
+    'should compile with cuda.',
 )
 class TestVmmCompact(unittest.TestCase):
     def setUp(self):