fix retry_allocator (#17245)

test=develop

fix retry_allocator (#17245)
test=develop
6fafd37e · Zeng Jinle · GitHub · ff1661f1 · 6fafd37e
隐藏空白更改
内联并排

Showing with 4 addition and 0 deletion

paddle/fluid/memory/allocation/CMakeLists.txt paddle/fluid/memory/allocation/CMakeLists.txt +4 -0

未找到文件。
--- a/paddle/fluid/memory/allocation/CMakeLists.txt
+++ b/paddle/fluid/memory/allocation/CMakeLists.txt
@@ -61,6 +61,10 @@ nv_test(allocation_and_eigen_test SRCS allocation_and_eigen_test.cu DEPS allocat

 cc_test(retry_allocator_test SRCS retry_allocator_test.cc DEPS retry_allocator best_fit_allocator locked_allocator cpu_allocator)

+if (WITH_TESTING)
+  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_facade)

 cc_test(allocator_facade_frac_flags_test SRCS allocator_facade_frac_flags_test.cc DEPS allocator_facade)