From 9f47eb39af9b58ce76940b4bbbb135ae7e3b503c Mon Sep 17 00:00:00 2001 From: xiayanming Date: Fri, 26 Nov 2021 22:37:01 +0800 Subject: [PATCH] fix ci issue --- python/paddle/distributed/fleet/launch.py | 10 ++-------- 1 file changed, 2 insertions(+), 8 deletions(-) diff --git a/python/paddle/distributed/fleet/launch.py b/python/paddle/distributed/fleet/launch.py index 38a197e49a893..e5d1a727cd483 100644 --- a/python/paddle/distributed/fleet/launch.py +++ b/python/paddle/distributed/fleet/launch.py @@ -287,21 +287,15 @@ def cpuonly_check(args): def get_cluster_info(args): # parse arguments, used for cloud-single-machine and local if args.backend == 'gloo': cpuonly_check(args) - if args.enable_auto_mapping: - (device_mode, devices_per_proc) = (DeviceMode.GPU, []) - else: - (device_mode, - devices_per_proc) = launch_utils.get_device_proc_info(args) + (device_mode, devices_per_proc) = launch_utils.get_device_proc_info(args) trainers_num = cloud_utils.get_trainers_num() logger.debug("parsed from args trainerss_num:{} mode:{} devices:{}".format( trainers_num, device_mode, devices_per_proc)) - cuda_visible_devices = os.getenv("CUDA_VISIBLE_DEVICES") - cluster = None pod = None - start_port = 6170 + start_port = 6170 if os.environ.get('FLAGS_START_PORT') is not None: start_port = os.environ.get('FLAGS_START_PORT')