Skip to content

Commit 6069d53

Browse files
committed
add local_gpu condition for warning ps
1 parent 69d06ad commit 6069d53

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/sagemaker/fw_utils.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -613,7 +613,7 @@ def warn_if_parameter_server_with_multi_gpu(training_instance_type, distribution
613613
614614
615615
"""
616-
if training_instance_type == "local" or distributions is None:
616+
if training_instance_type in ["local","local_gpu"] or distributions is None:
617617
return
618618

619619
is_multi_gpu_instance = (

0 commit comments

Comments
 (0)