Support aml in hybrid pipeline (#4124)

microsoft · Sep 6, 2021 · dfba3ae · dfba3ae
1 parent 31fa181
commit dfba3ae
Show file tree

Hide file tree

Showing 3 changed files with 14 additions and 1 deletion.
diff --git a/pipelines/integration-test-hybrid.yml b/pipelines/integration-test-hybrid.yml
@@ -85,6 +85,10 @@ jobs:
           --remote_pwd $(password_in_docker) \
           --remote_port $(docker_port) \
           --nni_manager_ip $(manager_ip) \
+          --subscription_id $(subscription_id) \
+          --resource_group $(resource_group) \
+          --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
     displayName: Integration test

diff --git a/test/config/training_service_v2.yml b/test/config/training_service_v2.yml
@@ -6,4 +6,9 @@ hybrid:
         user: 
         password: 
         port: 
-  - platform: local
+  - platform: local
+  - platform: aml
+    subscriptionId:
+    resourceGroup:
+    workspaceName:
+    computeTarget:
diff --git a/test/nni_test/nnitest/generate_ts_config.py b/test/nni_test/nnitest/generate_ts_config.py
@@ -120,6 +120,10 @@ def update_training_service_config(args):
         config[args.ts]['trainingService'][0]['machineList'][0]['host'] = args.remote_host
         config[args.ts]['trainingService'][0]['machineList'][0]['password'] = args.remote_pwd
         config[args.ts]['trainingService'][0]['machineList'][0]['port'] = args.remote_port
+        config[args.ts]['trainingService'][2]['subscriptionId'] = args.subscription_id
+        config[args.ts]['trainingService'][2]['resourceGroup'] = args.resource_group
+        config[args.ts]['trainingService'][2]['workspaceName'] = args.workspace_name
+        config[args.ts]['trainingService'][2]['computeTarget'] = args.compute_target
         config[args.ts]['nni_manager_ip'] = args.nni_manager_ip
         dump_yml_content(TRAINING_SERVICE_FILE_V2, config)