Refactor integration test (step 7) - pipeline updates (linux part) (#4898)

dependencies/develop.txt

@@ -10,7 +10,7 @@ pylint
 pyright == 1.1.250
 pytest
 pytest-cov
-rstcheck
+rstcheck >= 6.0
 sphinx >= 4.5
 sphinx-argparse-nni >= 0.4.0
 sphinx-copybutton

docs/source/deprecated/oneshot_legacy.rst

@@ -172,19 +172,19 @@ To run the tutorial, follow the steps below:
 
 2. **Search**: Based on the architecture of simplified PFLD, the setting of multi-stage search space and hyper-parameters for searching should be firstly configured to construct the supernet. For example,
 
-   .. code-block::
+   .. code-block:: python
 
       from lib.builder import search_space
       from lib.ops import PRIMITIVES
       from lib.supernet import PFLDInference, AuxiliaryNet
-      from nni.algorithms.nas.pytorch.fbnet import LookUpTable, NASConfig,
+      from nni.algorithms.nas.pytorch.fbnet import LookUpTable, NASConfig
 
       # configuration of hyper-parameters
       # search_space defines the multi-stage search space
       nas_config = NASConfig(
-         model_dir=^./ckpt_save^,
+         model_dir="./ckpt_save",
          nas_lr=0.01,
-         mode=^mul^,
+         mode="mul",
          alpha=0.25,
          beta=0.6,
          search_space=search_space,

pipelines/fast-test.yml

@@ -24,7 +24,7 @@ stages:
         rstcheck -r source \
           --ignore-directives automodule,autoclass,autofunction,cardlinkitem,codesnippetcard,argparse,tabs \
           --ignore-roles githublink --ignore-substitutions release \
-          --report warning
+          --report-level warning
       displayName: rstcheck
 
     - script: |

pipelines/full-test-compression.yml

@@ -11,15 +11,10 @@ jobs:
   timeoutInMinutes: 120
 
   steps:
-  - template: templates/setup-non-ms-hosted-agent.yml
-
-  - template: templates/config-version.yml
-    parameters:
-      platform: ubuntu-latest-gpu
-
   - template: templates/install-dependencies.yml
     parameters:
       platform: ubuntu-latest-gpu
+      python_env: venv
 
   - template: templates/install-nni.yml
 

pipelines/full-test-hpo.yml

@@ -11,15 +11,10 @@ jobs:
   timeoutInMinutes: 120
 
   steps:
-  - template: templates/setup-non-ms-hosted-agent.yml
-
-  - template: templates/config-version.yml
-    parameters:
-      platform: ubuntu-latest-gpu
-
   - template: templates/install-dependencies.yml
     parameters:
       platform: ubuntu-latest-gpu
+      python_env: venv
 
   - template: templates/install-nni.yml
 
@@ -42,7 +37,7 @@ jobs:
 
   - script: |
       cd test
-      python nni_test/nnitest/run_tests.py --config config/integration_tests.yml --ts local
+      python training_service/nnitest/run_tests.py --config training_service/config/integration_tests.yml --ts local
     displayName: Integration test
 
 # TODO: should add a test on platforms other than linux

pipelines/full-test-nas.yml

@@ -11,15 +11,10 @@ jobs:
   timeoutInMinutes: 120
 
   steps:
-  - template: templates/setup-non-ms-hosted-agent.yml
-
-  - template: templates/config-version.yml
-    parameters:
-      platform: ubuntu-latest-gpu
-
   - template: templates/install-dependencies.yml
     parameters:
       platform: ubuntu-latest-gpu
+      python_env: venv
 
   - template: templates/install-nni.yml
 

pipelines/integration-test-adl.yml

@@ -53,11 +53,11 @@ jobs:
   - script: |
       set -e
       cd test
-      python3 nni_test/nnitest/generate_ts_config.py \
+      python3 training_service/nnitest/generate_ts_config.py \
           --ts adl \
           --nni_docker_image nnidev/nni-nightly \
           --checkpoint_storage_class $(checkpoint_storage_class) \
           --checkpoint_storage_size $(checkpoint_storage_size) \
           --nni_manager_ip $(nni_manager_ip)
-      python3 nni_test/nnitest/run_tests.py --config config/integration_tests.yml --ts adl --exclude multi-phase,multi-thread
+      python3 training_service/nnitest/run_tests.py --config training_service/config/integration_tests.yml --ts adl --exclude multi-phase,multi-thread
     displayName: Integration test

pipelines/integration-test-aml.yml

@@ -7,35 +7,23 @@ schedules:
 
 jobs:
 - job: aml
-  pool: NNI CI REMOTE CLI
+  pool:
+    vmImage: ubuntu-latest
   timeoutInMinutes: 120
 
   steps:
-  - script: |
-      export NNI_RELEASE=999.$(date -u +%Y%m%d%H%M%S)
-      echo "##vso[task.setvariable variable=PATH]${PATH}:${HOME}/.local/bin"
-      echo "##vso[task.setvariable variable=NNI_RELEASE]${NNI_RELEASE}"
+  - template: templates/install-dependencies.yml
+    parameters:
+      platform: ubuntu-latest
 
-      echo "Working directory: ${PWD}"
-      echo "NNI version: ${NNI_RELEASE}"
-      echo "Build docker image: $(build_docker_image)"
+  - template: templates/install-dependencies-aml.yml
 
-      python3 -m pip install --upgrade pip setuptools
-    displayName: Prepare
+  - template: templates/install-nni.yml
+    parameters:
+      wheel: true
+      extra_dep: SMAC,BOHB
 
-  - script: |
-      set -e
-      python3 setup.py build_ts
-      python3 setup.py bdist_wheel -p manylinux1_x86_64
-      python3 -m pip install dist/nni-${NNI_RELEASE}-py3-none-manylinux1_x86_64.whl[SMAC,BOHB]
-    displayName: Build and install NNI
-
-  - script: |
-      set -e
-      cd examples/tuners/customized_tuner
-      python3 setup.py develop --user
-      nnictl algo register --meta meta_file.yml
-    displayName: Install customized tuner
+  - template: templates/install-customized-tuner.yml
 
   - script: |
       set -e
@@ -50,8 +38,7 @@ jobs:
   - script: |
       set -e
       cd test
-      az login --service-principal -u $(client_id) -p $(client_secret) --tenant $(tenant_id)
-      python3 nni_test/nnitest/generate_ts_config.py \
+      python training_service/nnitest/generate_ts_config.py \
           --ts aml \
           --subscription_id $(subscriptionId) \
           --resource_group $(resourceGroup) \
@@ -59,6 +46,7 @@ jobs:
           --compute_target $(computeTarget) \
           --nni_manager_ip $(manager_ip) \
           --nni_docker_image nnidev/nni-nightly
-
-      python3 nni_test/nnitest/run_tests.py --config config/integration_tests.yml --ts aml
+      python training_service/nnitest/run_tests.py --config training_service/config/integration_tests.yml --ts aml
     displayName: Integration test
+
+  - template: templates/save-crashed-info.yml

pipelines/integration-test-frameworkcontroller.yml

@@ -40,7 +40,7 @@ jobs:
   - script: |
       set -e
       cd test
-      python3 nni_test/nnitest/generate_ts_config.py \
+      python3 training_service/nnitest/generate_ts_config.py \
           --ts frameworkcontroller \
           --keyvault_vaultname $(keyvault_vaultname) \
           --keyvault_name $(keyvault_name) \
@@ -48,13 +48,13 @@ jobs:
           --azs_share $(azs_share) \
           --nni_docker_image nnidev/nni-nightly \
           --nni_manager_ip $(manager_ip)
-      python3 nni_test/nnitest/run_tests.py --config config/integration_tests.yml --ts frameworkcontroller --exclude multi-phase,multi-thread
+      python3 training_service/nnitest/run_tests.py --config training_service/config/integration_tests.yml --ts frameworkcontroller --exclude multi-phase,multi-thread
     displayName: Integration test
 
   - script: |
       set -e
       cd test
-      python3 nni_test/nnitest/generate_ts_config.py \
+      python3 training_service/nnitest/generate_ts_config.py \
           --ts frameworkcontroller \
           --keyvault_vaultname  $(keyvault_vaultname) \
           --keyvault_name $(keyvault_name) \
@@ -64,5 +64,5 @@ jobs:
           --nni_manager_ip $(manager_ip) \
           --reuse_mode True \
           --config_version v2
-      python3 nni_test/nnitest/run_tests.py --config config/integration_tests_config_v2.yml --ts frameworkcontroller --reuse_mode True --exclude multi-phase,multi-thread
+      python3 training_service/nnitest/run_tests.py --config training_service/config/integration_tests_config_v2.yml --ts frameworkcontroller --reuse_mode True --exclude multi-phase,multi-thread
     displayName: Integration test (reuse mode)

pipelines/integration-test-hybrid.yml

@@ -7,77 +7,56 @@ schedules:
 
 jobs:
 - job: hybrid
-  pool: NNI CI REMOTE CLI
+  pool: nni-it
   timeoutInMinutes: 120
 
   steps:
+  # FIXME: should use GPU here
+
+  - template: templates/install-dependencies.yml
+    parameters:
+      platform: ubuntu-latest
+      python_env: venv
+
+  - template: templates/install-dependencies-aml.yml
+
+  - template: templates/install-nni.yml
+    parameters:
+      wheel: true
+      extra_dep: SMAC,BOHB
+
+  - template: templates/install-customized-tuner.yml
+
+  # These parts are exactly same as remote.
+  # Refactor these as templates when the refactor is done.
   - script: |
-      export NNI_RELEASE=999.$(date -u +%Y%m%d%H%M%S)
-      echo "##vso[task.setvariable variable=PATH]${PATH}:${HOME}/.local/bin"
-      echo "##vso[task.setvariable variable=NNI_RELEASE]${NNI_RELEASE}"
-
-      echo "Working directory: ${PWD}"
-      echo "NNI version: ${NNI_RELEASE}"
-
-      python3 -m pip install --upgrade pip setuptools
-    displayName: Prepare
+      set -e
+      mkdir -p /tmp/nnitest/$(Build.BuildId)
+      cp Dockerfile /tmp/nnitest/$(Build.BuildId)
+      cp -r dist /tmp/nnitest/$(Build.BuildId)
+      cp -r test /tmp/nnitest/$(Build.BuildId)
+      cp -r test/vso_tools/interim_patch.py /tmp/nnitest/$(Build.BuildId)
+    displayName: Prepare for docker
 
   - script: |
       set -e
-      python3 test/vso_tools/install_nni.py $(NNI_RELEASE) SMAC,BOHB
+      export worker_ip=localhost
+      export manager_ip=host.docker.internal
+      export password_in_docker=123456
+      echo "##vso[task.setvariable variable=worker_ip]${worker_ip}"
+      echo "##vso[task.setvariable variable=manager_ip]${manager_ip}"
+      echo "##vso[task.setvariable variable=password_in_docker]${password_in_docker}"
+    displayName: Override environment variables
 
-      cd examples/tuners/customized_tuner
-      python3 setup.py develop --user
-      nnictl algo register --meta meta_file.yml
-    displayName: Install NNI
-
-  - task: CopyFilesOverSSH@0
-    inputs:
-      sshEndpoint: $(worker)
-      sourceFolder: dist
-      targetFolder: /tmp/nnitest/$(Build.BuildId)/dist
-      overwrite: true
-    displayName: Copy wheel to remote machine
-    timeoutInMinutes: 10
-
-  - task: CopyFilesOverSSH@0
-    inputs:
-      sshEndpoint: $(worker)
-      contents: Dockerfile
-      targetFolder: /tmp/nnitest/$(Build.BuildId)
-      overwrite: true
-    displayName: Copy dockerfile to remote machine
-    timeoutInMinutes: 10
-
-  - task: CopyFilesOverSSH@0
-    inputs:
-      sshEndpoint: $(worker)
-      sourceFolder: test
-      targetFolder: /tmp/nnitest/$(Build.BuildId)/test
-      overwrite: true
-    displayName: Copy test scripts to remote machine
-    timeoutInMinutes: 10
-
-  # Need del later
-  - task: CopyFilesOverSSH@0
-    inputs:
-      sshEndpoint: $(worker)
-      contents: test/vso_tools/interim_patch.py
-      targetFolder: /tmp/nnitest/$(Build.BuildId)
-      overwrite: true
-    displayName: Copy torch patch to remote machine
-    timeoutInMinutes: 10
-
-  - task: SSH@0
-    inputs:
-      sshEndpoint: $(worker)
-      runOptions: commands
-      commands: python3 /tmp/nnitest/$(Build.BuildId)/test/vso_tools/start_docker.py $(NNI_RELEASE) $(Build.BuildId) $(password_in_docker)
-    displayName: Install NNI and run docker on Linux worker
+  - script: |
+      set -e
+      python /tmp/nnitest/$(Build.BuildId)/test/vso_tools/start_docker.py --sudo $(NNI_RELEASE) $(Build.BuildId) $(password_in_docker)
+    displayName: Run a worker in docker
+  # End of duplicated region
 
   - script: |
       cd test
-      python3 nni_test/nnitest/generate_ts_config.py \
+      python training_service/nnitest/generate_ts_config.py \
           --ts hybrid \
           --remote_reuse true \
           --remote_user nni \
@@ -90,13 +69,7 @@ jobs:
           --workspace_name $(workspace_name) \
           --compute_target $(compute_target) \
           --config_version v2
-      python3 nni_test/nnitest/run_tests.py --config config/integration_tests_config_v2.yml --ts hybrid
+      python training_service/nnitest/run_tests.py --config training_service/config/integration_tests_config_v2.yml --ts hybrid
     displayName: Integration test
 
-  - task: SSH@0
-    inputs:
-      sshEndpoint: $(worker)
-      runOptions: commands
-      commands: python3 /tmp/nnitest/$(Build.BuildId)/test/vso_tools/stop_docker.py $(Build.BuildId)
-    condition: always()
-    displayName: Stop docker
+  - template: templates/save-crashed-info.yml

pipelines/integration-test-kubeflow.yml

@@ -51,7 +51,7 @@ jobs:
       set -e
       cd test
       az login --service-principal -u $(client_id) -p $(client_secret) --tenant $(tenant_id)
-      python3 nni_test/nnitest/generate_ts_config.py \
+      python3 training_service/nnitest/generate_ts_config.py \
           --ts kubeflow \
           --keyvault_vaultname $(keyvault_vaultname) \
           --keyvault_name $(keyvault_name) \
@@ -59,14 +59,14 @@ jobs:
           --azs_share $(azs_share) \
           --nni_docker_image nnidev/nni-nightly \
           --nni_manager_ip $(manager_ip)
-      python3 nni_test/nnitest/run_tests.py --config config/integration_tests.yml --ts kubeflow --exclude multi-phase,multi-thread
+      python3 training_service/nnitest/run_tests.py --config training_service/config/integration_tests.yml --ts kubeflow --exclude multi-phase,multi-thread
     displayName: Integration test
 
   - script: |
       set -e
       cd test
       az login --service-principal -u $(client_id) -p $(client_secret) --tenant $(tenant_id)
-      python3 nni_test/nnitest/generate_ts_config.py \
+      python3 training_service/nnitest/generate_ts_config.py \
           --ts kubeflow \
           --keyvault_vaultname $(keyvault_vaultname) \
           --keyvault_name $(keyvault_name) \
@@ -76,5 +76,5 @@ jobs:
           --nni_manager_ip $(manager_ip) \
           --reuse_mode True \
           --config_version v2
-      python3 nni_test/nnitest/run_tests.py --config config/integration_tests_config_v2.yml --ts kubeflow --reuse_mode True --exclude multi-phase,multi-thread
+      python3 training_service/nnitest/run_tests.py --config training_service/config/integration_tests_config_v2.yml --ts kubeflow --reuse_mode True --exclude multi-phase,multi-thread
     displayName: Integration test (reuse mode)

pipelines/integration-test-local-linux.yml

@@ -0,0 +1,28 @@
+trigger: none
+pr: none
+schedules:
+- cron: 0 16 * * *
+  branches:
+    include: [ master ]
+
+jobs:
+- job: linux
+  pool: nni-it
+  timeoutInMinutes: 120
+
+  steps:
+  - template: templates/install-dependencies.yml
+    parameters:
+      platform: ubuntu-latest-gpu
+      python_env: venv
+
+  - template: templates/install-nni.yml
+
+  - template: templates/install-customized-tuner.yml
+
+  - script: |
+      cd test
+      python training_service/nnitest/run_tests.py --config training_service/config/integration_tests.yml --ts local
+    displayName: Integration test
+
+  - template: templates/save-crashed-info.yml

pipelines/integration-test-local.yml

@@ -1,75 +0,0 @@
-trigger: none
-pr: none
-schedules:
-- cron: 0 16 * * *
-  branches:
-    include: [ master ]
-
-jobs:
-- job: linux
-  pool: nni-it
-  timeoutInMinutes: 120
-
-  steps:
-  - template: templates/setup-non-ms-hosted-agent.yml
-
-  - template: templates/config-version.yml
-    parameters:
-      platform: ubuntu-latest-gpu
-
-  - template: templates/install-dependencies.yml
-    parameters:
-      platform: ubuntu-latest-gpu
-
-  - template: templates/install-nni.yml
-
-  - script: |
-      set -e
-      cd examples/tuners/customized_tuner
-      python setup.py develop
-      nnictl algo register --meta meta_file.yml
-    displayName: Install customized tuner
-
-  - script: |
-      cd test
-      python nni_test/nnitest/run_tests.py --config config/integration_tests.yml --ts local
-    displayName: Integration test
-
-- job: windows
-  pool: NNI CI WINDOWS FULL TEST
-  timeoutInMinutes: 120
-
-  steps:
-  - script: |
-      python -m pip install -U -r dependencies/setup.txt
-      python -m pip install -r dependencies/develop.txt
-      python -m pip install -r dependencies/recommended.txt
-    displayName: Install Python tools
-
-  - script: |
-      python -m pip uninstall nni --yes
-      set NNI_RELEASE=999.0
-      python setup.py build_ts
-      python setup.py bdist_wheel -p win_amd64
-      python -m pip install dist/nni-999.0-py3-none-win_amd64.whl[PPOTuner,DNGO]
-    displayName: Install NNI
-
-  - script: |
-      cd examples/tuners/customized_tuner
-      python setup.py develop --user
-      nnictl algo register --meta meta_file.yml
-    displayName: Install example customized tuner
-
-  # TODO: remove unit test here
-
-  - script: |
-      cd test
-      python -m pytest ut
-      echo "TODO: TypeScript UT"
-    displayName: Unit test
-    continueOnError: true
-
-  - script: |
-      cd test
-      python nni_test/nnitest/run_tests.py --config config/integration_tests.yml --ts local
-    displayName: Integration test

pipelines/integration-test-openpai-linux.yml

@@ -56,7 +56,7 @@ jobs:
   - script: |
       set -e
       cd test
-      python3 nni_test/nnitest/generate_ts_config.py \
+      python3 training_service/nnitest/generate_ts_config.py \
           --ts pai \
           --pai_reuse false \
           --pai_host https://ne.openpai.org \
@@ -69,13 +69,13 @@ jobs:
           --nni_manager_ip $(manager_ip) \
           --vc nni \
           --debug true
-      python3 nni_test/nnitest/run_tests.py --config config/integration_tests.yml --ts pai
+      python3 training_service/nnitest/run_tests.py --config training_service/config/integration_tests.yml --ts pai
     displayName: Integration test
 
   - script: |
       set -e
       cd test
-      python3 nni_test/nnitest/generate_ts_config.py \
+      python3 training_service/nnitest/generate_ts_config.py \
           --ts pai \
           --pai_reuse true \
           --pai_host https://ne.openpai.org \
@@ -87,5 +87,5 @@ jobs:
           --container_nfs_mount_path $(container_nfs_mount_path) \
           --nni_manager_ip $(manager_ip) \
           --vc nni
-      python3 nni_test/nnitest/run_tests.py --config config/integration_tests.yml --ts pai
+      python3 training_service/nnitest/run_tests.py --config training_service/config/integration_tests.yml --ts pai
     displayName: Integration test (reuse mode)

pipelines/integration-test-remote-l2l.yml

@@ -1,9 +1,9 @@
 trigger: none
 pr: none
-schedules:
-- cron: 0 16 * * *
-  branches:
-    include: [ master ]
+# schedules:
+# - cron: 0 16 * * *
+#   branches:
+#     include: [ master ]
 
 # variables set on VSO: (for security concern)
 #   manager_ip
@@ -12,77 +12,61 @@ schedules:
 
 jobs:
 - job: remote_linux2linux
-  pool: NNI CI REMOTE CLI
+  pool: nni-it
   timeoutInMinutes: 140
 
   steps:
+  # FIXME: GPU is not supported yet.
+  # Change to ubuntu-latest-gpu when it's done.
+
+  - template: templates/install-dependencies.yml
+    parameters:
+      platform: ubuntu-latest
+      python_env: venv
+
+  - template: templates/install-nni.yml
+    parameters:
+      wheel: true
+      extra_dep: SMAC,BOHB
+
+  - template: templates/install-customized-tuner.yml
+
   - script: |
-      export NNI_RELEASE=999.$(date -u +%Y%m%d%H%M%S)
-      echo "##vso[task.setvariable variable=PATH]${PATH}:${HOME}/.local/bin"
-      echo "##vso[task.setvariable variable=NNI_RELEASE]${NNI_RELEASE}"
-
-      echo "Working directory: ${PWD}"
-      echo "NNI version: ${NNI_RELEASE}"
-
-      python3 -m pip install --upgrade pip setuptools
-    displayName: Prepare
+      sudo systemctl enable nfs-kernel-server
+    displayName: Enable NFS service
 
   - script: |
       set -e
-      python3 test/vso_tools/install_nni.py $(NNI_RELEASE) SMAC,BOHB
+      mkdir -p /tmp/nnitest/$(Build.BuildId)
+      cp Dockerfile /tmp/nnitest/$(Build.BuildId)
+      cp -r dist /tmp/nnitest/$(Build.BuildId)
+      cp -r test /tmp/nnitest/$(Build.BuildId)
+      cp -r test/vso_tools/interim_patch.py /tmp/nnitest/$(Build.BuildId)
+    displayName: Prepare for docker
 
-      cd examples/tuners/customized_tuner
-      python3 setup.py develop --user
-      nnictl algo register --meta meta_file.yml
-    displayName: Install NNI
+  # TODO: Delete variables on pipeline when the PR that introduced this is merged.
+  - script: |
+      set -e
+      export worker_ip=localhost
+      export manager_ip=host.docker.internal
+      export password_in_docker=123456
+      export NFS_IP=host.docker.internal
+      export Exported_Directory=/var/nfs/general
+      echo "##vso[task.setvariable variable=worker_ip]${worker_ip}"
+      echo "##vso[task.setvariable variable=manager_ip]${manager_ip}"
+      echo "##vso[task.setvariable variable=password_in_docker]${password_in_docker}"
+      echo "##vso[task.setvariable variable=NFS_IP]${NFS_IP}"
+      echo "##vso[task.setvariable variable=Exported_Directory]${Exported_Directory}"
+    displayName: Override environment variables
 
-  - task: CopyFilesOverSSH@0
-    inputs:
-      sshEndpoint: $(worker)
-      sourceFolder: dist
-      targetFolder: /tmp/nnitest/$(Build.BuildId)/dist
-      overwrite: true
-    displayName: Copy wheel to remote machine
-    timeoutInMinutes: 10
-
-  - task: CopyFilesOverSSH@0
-    inputs:
-      sshEndpoint: $(worker)
-      contents: Dockerfile
-      targetFolder: /tmp/nnitest/$(Build.BuildId)
-      overwrite: true
-    displayName: Copy dockerfile to remote machine
-    timeoutInMinutes: 10
-
-  - task: CopyFilesOverSSH@0
-    inputs:
-      sshEndpoint: $(worker)
-      sourceFolder: test
-      targetFolder: /tmp/nnitest/$(Build.BuildId)/test
-      overwrite: true
-    displayName: Copy test scripts to remote machine
-    timeoutInMinutes: 10
-
-  # Need del later
-  - task: CopyFilesOverSSH@0
-    inputs:
-      sshEndpoint: $(worker)
-      contents: test/vso_tools/interim_patch.py
-      targetFolder: /tmp/nnitest/$(Build.BuildId)
-      overwrite: true
-    displayName: Copy torch patch to remote machine
-    timeoutInMinutes: 10
-
-  - task: SSH@0
-    inputs:
-      sshEndpoint: $(worker)
-      runOptions: commands
-      commands: python3 /tmp/nnitest/$(Build.BuildId)/test/vso_tools/start_docker.py $(NNI_RELEASE) $(Build.BuildId) $(password_in_docker)
-    displayName: Install NNI and run docker on Linux worker
+  - script: |
+      set -e
+      python /tmp/nnitest/$(Build.BuildId)/test/vso_tools/start_docker.py --sudo $(NNI_RELEASE) $(Build.BuildId) $(password_in_docker)
+    displayName: Run a worker in docker
 
   - script: |
       cd test
-      python3 nni_test/nnitest/generate_ts_config.py \
+      python training_service/nnitest/generate_ts_config.py \
           --ts remote \
           --remote_reuse true \
           --remote_user nni \
@@ -95,12 +79,12 @@ jobs:
           --local_mount_point $(LOCAL_MOUNT_POINT) \
           --remote_mount_point $(REMOTE_MOUNT_POINT) \
           --exported_directory $(Exported_Directory)
-      python3 nni_test/nnitest/run_tests.py --config config/integration_tests.yml --ts remote
+      python training_service/nnitest/run_tests.py --config training_service/config/integration_tests.yml --ts remote
     displayName: Integration test (reuse mode)
 
   - script: |
       cd test
-      python3 nni_test/nnitest/generate_ts_config.py \
+      python training_service/nnitest/generate_ts_config.py \
           --ts remote \
           --remote_reuse false \
           --remote_user nni \
@@ -108,14 +92,9 @@ jobs:
           --remote_port $(docker_port) \
           --remote_pwd $(password_in_docker) \
           --nni_manager_ip $(manager_ip)
-      python3 nni_test/nnitest/run_tests.py --config config/integration_tests.yml --ts remote
+      python training_service/nnitest/run_tests.py --config training_service/config/integration_tests.yml --ts remote
     displayName: Integration test
 
-
-  - task: SSH@0
-    inputs:
-      sshEndpoint: $(worker)
-      runOptions: commands
-      commands: python3 /tmp/nnitest/$(Build.BuildId)/test/vso_tools/stop_docker.py $(Build.BuildId)
-    condition: always()
-    displayName: Stop docker
+  - template: templates/save-crashed-info.yml
+    parameters:
+      remote: true

pipelines/integration-test-trt.yml

@@ -33,5 +33,5 @@ jobs:
   - script: |
       set -e
       cd test
-      python3 nni_test/nnitest/test_quantize_model_speedup.py
+      python3 training_service/nnitest/test_quantize_model_speedup.py
     displayName: Quantize model speedup test

test/vso_tools/start_docker.py

@@ -5,9 +5,10 @@ Usage:
     python start_docker.py <nni-version> <container-name> <password-in-docker>
 """
 
+import argparse
+import os
 import random
 import socket
-import sys
 
 from _common import build_wheel, run_command, set_variable
 
@@ -20,14 +21,28 @@ while True:
     sock.close()
     port = random.randint(10000, 20000)
 
-version = sys.argv[1]
-container = sys.argv[2]
-password = sys.argv[3]
+def main():
+    parser = argparse.ArgumentParser()
+    parser.add_argument('version', type=str)
+    parser.add_argument('container', type=str)
+    parser.add_argument('password', type=str)
+    parser.add_argument('--sudo', default=False, action='store_true')
 
-run_command(f'docker build --build-arg NNI_RELEASE={version} -t nnidev/nni-nightly .')
-run_command(f'docker run --privileged -d -t -p {port}:22 --name {container} nnidev/nni-nightly')
-run_command(f'docker exec {container} useradd --create-home --password {password} nni')
-run_command(['docker', 'exec', container, 'bash', '-c', f'echo "nni:{password}" | chpasswd'])
-run_command(['docker', 'exec', container, 'bash', '-c', 'echo "nni ALL=(ALL:ALL) NOPASSWD:ALL" >> /etc/sudoers'])
-run_command(f'docker exec {container} service ssh start')
-set_variable('docker_port', port)
+    args = parser.parse_args()
+    docker = 'sudo docker' if args.sudo else 'docker'
+    version, container, password = args.version, args.container, args.password
+    uid, gid = os.getuid(), os.getgid()
+
+    run_command(f'{docker} build --build-arg NNI_RELEASE={version} -t nnidev/nni-nightly .')
+    run_command(f'{docker} run --privileged -d -t -p {port}:22 --add-host=host.docker.internal:host-gateway --name {container} nnidev/nni-nightly')
+    # The user inside docker must have the same uid and gid as outside.
+    # Otherwise NFS will have permission errors.
+    run_command(f'{docker} exec {container} groupadd -g {gid} nni')
+    run_command(f'{docker} exec {container} useradd --create-home --password {password} -u {uid} -g {gid} nni')
+    run_command(docker.split() + ['exec', container, 'bash', '-c', f'echo "nni:{password}" | chpasswd'])
+    run_command(docker.split() + ['exec', container, 'bash', '-c', 'echo "nni ALL=(ALL:ALL) NOPASSWD:ALL" >> /etc/sudoers'])
+    run_command(f'{docker} exec {container} service ssh start')
+    set_variable('docker_port', port)
+
+if __name__ == '__main__':
+    main()