Skip to content

Commit 69b3956

Browse files
authored
Merge pull request #2091 from FedML-AI/alaydshah/gpu_utils/qualcomm3
Nit: Update naming
2 parents 2bbe2f9 + 2fcf57d commit 69b3956

1 file changed

Lines changed: 5 additions & 4 deletions

File tree

python/fedml/computing/scheduler/model_scheduler/device_model_deployment.py

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -305,15 +305,16 @@ def start_deployment(end_point_id, end_point_name, model_id, model_version,
305305
# TODO: Make sure no competition for each replica in a single deployment
306306
if exist_container_obj is not None:
307307
client.api.remove_container(exist_container_obj.id, v=True, force=True)
308-
device_requests = {}
308+
device_mapping = {}
309309
if no_real_gpu_allocation is not None:
310310
use_gpu = not no_real_gpu_allocation
311311
if use_gpu:
312312
logging.info("Number of GPUs: {}".format(num_gpus))
313-
device_requests = HardwareUtil.get_docker_gpu_device_mapping(gpu_ids, num_gpus)
314-
logging.info(f"device_requests: {device_requests}")
313+
device_mapping = HardwareUtil.get_docker_gpu_device_mapping(gpu_ids, num_gpus)
314+
logging.info(f"device_mapping: {device_mapping}")
315315

316-
host_config_dict.update(device_requests)
316+
if device_mapping:
317+
host_config_dict.update(device_mapping)
317318

318319
# Environment variables
319320
if not enable_custom_image:

0 commit comments

Comments
 (0)