fix some static manual dist uts (#67902)

ooooo-create · web-flow · commit 99cf4b14bc25 · 2024-09-04T17:00:32.000+08:00
diff --git a/test/deprecated/asp/CMakeLists.txt b/test/deprecated/asp/CMakeLists.txt
@@ -4,16 +4,13 @@ file(
   "test_*.py")
 string(REPLACE ".py" "" TEST_OPS "${TEST_OPS}")
 
-list(REMOVE_ITEM TEST_OPS "test_fleet_with_asp_static_deprecated")
 list(REMOVE_ITEM TEST_OPS "test_fleet_with_asp_dynamic_deprecated")
 list(REMOVE_ITEM TEST_OPS "test_fleet_with_asp_sharding_deprecated")
 
 if(WITH_DISTRIBUTE)
   if(WITH_GPU OR WITH_XPU)
     py_test_modules(test_fleet_with_asp_dynamic_deprecated MODULES
                     test_fleet_with_asp_dynamic_deprecated ENVS ${dist_ENVS})
-    py_test_modules(test_fleet_with_asp_static_deprecated MODULES
-                    test_fleet_with_asp_static_deprecated ENVS ${dist_ENVS})
   endif()
 endif()
 
diff --git a/test/deprecated/asp/test_fleet_with_asp_static_deprecated.py b/test/deprecated/asp/test_fleet_with_asp_static_deprecated.py
diff --git a/test/deprecated/legacy_test/CMakeLists.txt b/test/deprecated/legacy_test/CMakeLists.txt
@@ -30,7 +30,6 @@ set(MIXED_DIST_TEST_OPS ${DIST_TEST_OPS})
 #remove distribute unittests.
 
 list(APPEND MIXED_DIST_TEST_OPS test_simple_dist_transpiler)
-list(APPEND MIXED_DIST_TEST_OPS test_communicator_async)
 list(APPEND MIXED_DIST_TEST_OPS test_communicator_ps_gpu)
 list(APPEND MIXED_DIST_TEST_OPS test_communicator_geo_deprecated)
 list(APPEND MIXED_DIST_TEST_OPS test_fleet_launch_ascend)
@@ -513,8 +512,6 @@ if(WITH_DISTRIBUTE)
     list(REMOVE_ITEM DIST_TEST_OPS "test_communicator_ps_gpu")
   endif()
 
-  py_test_modules(test_communicator_async MODULES test_communicator_async ENVS
-                  ${dist_ENVS})
   py_test_modules(test_communicator_geo_deprecated MODULES
                   test_communicator_geo_deprecated ENVS ${dist_ENVS})
   if(NOT APPLE)
diff --git a/test/deprecated/legacy_test/test_communicator_async.py b/test/deprecated/legacy_test/test_communicator_async.py
diff --git a/test/legacy_test/test_dist_base.py b/test/legacy_test/test_dist_base.py
@@ -15,6 +15,9 @@
 import argparse
 import ast
 import os
+
+os.environ['FLAGS_enable_pir_api'] = '0'
+
 import pickle
 import random
 import socket
diff --git a/test/legacy_test/test_dist_fleet_base.py b/test/legacy_test/test_dist_fleet_base.py
@@ -18,6 +18,8 @@
 
 import argparse
 import os
+
+os.environ['FLAGS_enable_pir_api'] = '0'
 import shutil
 import socket
 import subprocess
diff --git a/test/legacy_test/test_fleet_base_single.py b/test/legacy_test/test_fleet_base_single.py
@@ -14,6 +14,7 @@
 
 import os
 
+os.environ['FLAGS_enable_pir_api'] = '0'
 import numpy as np
 
 cuda_visible_devices = os.getenv('CUDA_VISIBLE_DEVICES')
diff --git a/tools/parallel_UT_rule.py b/tools/parallel_UT_rule.py
@@ -237,7 +237,6 @@
     'test_program',
     'mmap_allocator_test',
     'test_reshape_transpose_matmul_mkldnn_fuse_pass',
-    'test_communicator_async',
     'test_downpoursgd_deprecated',
     'variable_test',
     'test_quantization_mkldnn_pass',
@@ -2014,7 +2013,6 @@
     'test_fleet_rolemaker_2',
     'test_distributed_strategy',
     'test_rnn_cudnn_params_packing',
-    'test_communicator_async',
     'brpc_utils_test',
     'test_analyzer_capi_pd_tensor',
     'test_recv_save_op',
diff --git a/tools/test_run_by_protobuf_3.py b/tools/test_run_by_protobuf_3.py
@@ -20,7 +20,6 @@
     'test_fleet_base2',
     'test_fleet_base3',
     'test_communicator_geo_deprecated',
-    'test_communicator_async',
     'test_dist_fleet_a_sync_optimizer_async',
     'test_dist_fleet_a_sync_optimizer_auto',
     'test_dist_fleet_a_sync_optimizer_sync',
