Unverified Commit dfba3aed authored by SparkSnail's avatar SparkSnail Committed by GitHub
Browse files

Support aml in hybrid pipeline (#4124)

parent 31fa1811
......@@ -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
......
......@@ -7,3 +7,8 @@ hybrid:
password:
port:
- platform: local
- platform: aml
subscriptionId:
resourceGroup:
workspaceName:
computeTarget:
......@@ -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)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment