Update ansible playbooks to suppress warnings

- Get ansible_user_dir from facts
- Get hostname from facts
- Update NODE_RANK expression

superbench/runner/playbooks/check_env.yaml

@@ -20,16 +20,16 @@
   hosts: all
   gather_facts: true
   vars:
-    workspace: '{{ ansible_user_dir }}/sb-workspace'
+    workspace: '{{ ansible_facts["user_dir"] }}/sb-workspace'
     container: sb-workspace
     skip_docker: '{{ no_docker | default(false) }}'
-    sb_nodes: '{{ hostvars.values() | map(attribute="ansible_hostname") | sort }}'
+    sb_nodes: '{{ hostvars.values() | map(attribute="ansible_facts.hostname") | sort }}'
     sb_env: |
       # sb env
       SB_WORKSPACE={{ workspace if skip_docker else '/root' }}
       # pytorch env
       NNODES={{ sb_nodes | length }}
-      NODE_RANK={{ lookup('ansible.utils.index_of', sb_nodes, 'eq', ansible_hostname) }}
+      NODE_RANK={{ sb_nodes.index(ansible_facts["hostname"]) }}
       MASTER_ADDR={{ sb_nodes | first }}
       MASTER_PORT=29500
       OMP_NUM_THREADS=1

superbench/runner/playbooks/deploy.yaml

@@ -70,7 +70,7 @@
   hosts: all
   gather_facts: false
   vars:
-    workspace: '{{ ansible_user_dir }}/sb-workspace'
+    workspace: '{{ ansible_facts["user_dir"] }}/sb-workspace'
     container: sb-workspace
   tasks:
     - name: Creating Workspace

superbench/runner/playbooks/fetch_results.yaml

@@ -2,12 +2,12 @@
   hosts: all
   gather_facts: true
   vars:
-    workspace: '{{ ansible_user_dir }}/sb-workspace'
+    workspace: '{{ ansible_facts["user_dir"] }}/sb-workspace'
   tasks:
     - name: Synchronize Output Directory
       ansible.posix.synchronize:
         mode: pull
         src: '{{ sb_output_dir if sb_output_dir.startswith("/") else workspace + "/" + sb_output_dir  }}/'
-        dest: '{{ absolute_output_dir }}/nodes/{{ ansible_hostname }}'
+        dest: '{{ absolute_output_dir }}/nodes/{{ ansible_facts["hostname"] }}'
         rsync_opts:
           - --exclude=nodes

superbench/runner/templates/ssh_config.j2

-{% for host in hostvars.values() | map(attribute='ansible_hostname') | sort %}
+{% for host in hostvars.values() | map(attribute='ansible_facts.hostname') | sort %}
 Host {{ host }}
     Port {{ ssh_port }}
     IdentityFile /root/.ssh/key