Skip to content

Commit

Permalink
fix ci issue
Browse files Browse the repository at this point in the history
  • Loading branch information
xymyeah committed Nov 26, 2021
1 parent 6ebc69c commit 9f47eb3
Showing 1 changed file with 2 additions and 8 deletions.
10 changes: 2 additions & 8 deletions python/paddle/distributed/fleet/launch.py
Original file line number Diff line number Diff line change
Expand Up @@ -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')

Expand Down

0 comments on commit 9f47eb3

Please sign in to comment.