diff --git a/paddle/fluid/platform/device_event_test.cc b/paddle/fluid/platform/device_event_test.cc index 92fe7c02bd0bd6e267ce269f0ad5f3bd8b888bb7..dd68c0e5642139e2808dabf003f5bae7a671a8c0 100644 --- a/paddle/fluid/platform/device_event_test.cc +++ b/paddle/fluid/platform/device_event_test.cc @@ -44,8 +44,6 @@ TEST(DeviceEvent, CUDA) { ASSERT_EQ(status, true); // case 2. test for event_recorder event.Record(context); - status = event.Query(); - ASSERT_EQ(status, false); // case 3. test for event_finisher event.Finish(); status = event.Query(); diff --git a/python/paddle/fluid/tests/unittests/CMakeLists.txt b/python/paddle/fluid/tests/unittests/CMakeLists.txt index 00ca71ef359e01aa246c8577fcbaa8c04cf06309..45a6c815728d1728b15090bb9eff120ca399a7eb 100755 --- a/python/paddle/fluid/tests/unittests/CMakeLists.txt +++ b/python/paddle/fluid/tests/unittests/CMakeLists.txt @@ -1356,7 +1356,7 @@ set_tests_properties(test_matmul_op PROPERTIES TIMEOUT 120) set_tests_properties(test_nearest_interp_v2_op PROPERTIES TIMEOUT 120) set_tests_properties(test_trilinear_interp_op PROPERTIES TIMEOUT 120) set_tests_properties(test_bicubic_interp_v2_op PROPERTIES TIMEOUT 120) -set_tests_properties(test_gather_op PROPERTIES TIMEOUT 120) +set_tests_properties(test_gather_op PROPERTIES TIMEOUT 180) set_tests_properties(test_static_save_load PROPERTIES TIMEOUT 250) set_tests_properties(test_pylayer_op PROPERTIES TIMEOUT 120) set_tests_properties(test_paddle_save_load_binary PROPERTIES TIMEOUT 120) diff --git a/python/paddle/fluid/tests/unittests/test_collective_api_base.py b/python/paddle/fluid/tests/unittests/test_collective_api_base.py index ef011d2ddc83fa9b5a34abc507065c9826d18d85..96f2d2bf4d504f911a40e2530b087f198f33de29 100644 --- a/python/paddle/fluid/tests/unittests/test_collective_api_base.py +++ b/python/paddle/fluid/tests/unittests/test_collective_api_base.py @@ -208,21 +208,15 @@ class TestDistBase(unittest.TestCase): with_gloo = '0' else: with_gloo = '1' - required_envs = { - "FLAGS_fraction_of_gpu_memory_to_use": "0.15", - "FLAGS_eager_delete_tensor_gb": "0.0", - "PATH": os.getenv("PATH"), - "PYTHONPATH": os.getenv("PYTHONPATH", ""), - "LD_LIBRARY_PATH": os.getenv("LD_LIBRARY_PATH", ""), - "LD_PRELOAD": os.getenv("LD_PRELOAD", ""), - "FLAGS_call_stack_level": "2", - "GLOG_v": "3", + required_envs = os.environ.copy() + additional_envs = { "NCCL_P2P_DISABLE": "1", "STATIC_MODE": static_mode, "PADDLE_WITH_GLOO": with_gloo, "BACKEND": backend, "PATH_ID": path_id } + required_envs.update(additional_envs) required_envs.update(need_envs) if check_error_log: required_envs["GLOG_v"] = "3" diff --git a/python/paddle/fluid/tests/unittests/test_gather_op.py b/python/paddle/fluid/tests/unittests/test_gather_op.py index 0c356f4bc38cd037ff6b31eca8f1fe69d5af824b..f76094c92eb8a4df38c0397cb4ec423dea6b6437 100644 --- a/python/paddle/fluid/tests/unittests/test_gather_op.py +++ b/python/paddle/fluid/tests/unittests/test_gather_op.py @@ -317,7 +317,7 @@ class API_TestDygraphGather(unittest.TestCase): return x = np.random.rand(226862, 256).astype("float32") - index = np.random.randint(0, 22682, size=(11859027)) + index = np.random.randint(0, 22682, size=(8859027)) def test_dygraph(): with fluid.dygraph.guard():