Added an option to turn off google DNS (#189)

Author: arjunsuresh

automation/script/docker_utils.py

@@ -109,7 +109,7 @@ def prepare_docker_inputs(input_params, docker_settings,
         keys += [
             "skip_run_cmd", "pre_run_cmds", "run_cmd_prefix", "all_gpus", "num_gpus", "device", "gh_token",
             "port_maps", "shm_size", "pass_user_id", "pass_user_group", "extra_run_args", "detached", "interactive",
-            "dt", "it", "use_host_group_id", "use_host_user_id", "keep_detached", "reuse_existing"
+            "dt", "it", "use_host_group_id", "use_host_user_id", "keep_detached", "reuse_existing", "use_google_dns"
         ]
     # Collect Dockerfile inputs
     docker_inputs = {

script/app-mlperf-inference/meta.yaml

@@ -1959,7 +1959,8 @@ docker:
   skip_run_cmd: 'no'
   shm_size: '32gb'
   interactive: True
-  extra_run_args: ' --dns 8.8.8.8 --dns 8.8.4.4 --cap-add SYS_ADMIN  --cap-add SYS_TIME --security-opt apparmor=unconfined --security-opt seccomp=unconfined'
+  extra_run_args: ' --cap-add SYS_ADMIN  --cap-add SYS_TIME --security-opt apparmor=unconfined --security-opt seccomp=unconfined'
+  use_google_dns: yes
   os: ubuntu
   user: mlcuser
   mlc_repo: mlcommons@mlperf-automations

script/run-docker-container/customize.py

@@ -181,6 +181,9 @@ def postprocess(i):
     if env.get('MLC_DOCKER_EXTRA_RUN_ARGS', '') != '':
         run_opts += env['MLC_DOCKER_EXTRA_RUN_ARGS']
 
+    if is_true(env.get('MLC_DOCKER_USE_GOOGLE_DNS', '')):
+        run_opts += ' --dns 8.8.8.8 --dns 8.8.4.4 '
+
     if env.get('MLC_CONTAINER_TOOL', '') == 'podman' and env.get(
             'MLC_PODMAN_MAP_USER_ID', '').lower() not in ["no", "0", "false"]:
         run_opts += " --userns=keep-id"

script/run-docker-container/meta.yaml

@@ -58,6 +58,7 @@ input_mapping:
   save_script: MLC_DOCKER_SAVE_SCRIPT
   script_tags: MLC_DOCKER_RUN_SCRIPT_TAGS
   shm_size: MLC_DOCKER_SHM_SIZE
+  use_google_dns: MLC_DOCKER_USE_GOOGLE_DNS
 
 new_env_keys:
   - 'MLC_DOCKER_CONTAINER_ID'