From fc977ce5dd9c7cdbb3a7725f00d7d3bace0e2bc6 Mon Sep 17 00:00:00 2001
From: Binyang Li <binyli@microsoft.com>
Date: Mon, 29 Apr 2024 11:54:34 +0800
Subject: [PATCH] Move pipeline to Azure org (#296)

Move multi-nodes pipeline to Azure org to meet the compliance
requirements. Remove default value for BASE_IMAGE. Not allowed to use
3rd party registry in Dockerfile directly.
---
 .azure-pipelines/multi-nodes-test.yml | 2 +-
 docker/base-dev-x.dockerfile          | 2 +-
 docker/base-x.dockerfile              | 2 +-
 3 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/.azure-pipelines/multi-nodes-test.yml b/.azure-pipelines/multi-nodes-test.yml
index 7c9d35094..bcf517569 100644
--- a/.azure-pipelines/multi-nodes-test.yml
+++ b/.azure-pipelines/multi-nodes-test.yml
@@ -34,7 +34,7 @@ jobs:
     name: SshKeyFile
     displayName: Download key file
     inputs:
-      secureFile: ssh.key
+      secureFile: mscclpp-ssh.key
 
   - task: Bash@3
     name: InstallPackages
diff --git a/docker/base-dev-x.dockerfile b/docker/base-dev-x.dockerfile
index d7f2166f1..d6236cd02 100644
--- a/docker/base-dev-x.dockerfile
+++ b/docker/base-dev-x.dockerfile
@@ -1,4 +1,4 @@
-ARG BASE_IMAGE=ghcr.io/microsoft/mscclpp/mscclpp:base-cuda12.1
+ARG BASE_IMAGE
 FROM ${BASE_IMAGE}
 
 LABEL maintainer="MSCCL++"
diff --git a/docker/base-x.dockerfile b/docker/base-x.dockerfile
index bf29f718a..4be89c9d0 100644
--- a/docker/base-x.dockerfile
+++ b/docker/base-x.dockerfile
@@ -1,4 +1,4 @@
-ARG BASE_IMAGE=nvidia/cuda:12.1.1-devel-ubuntu20.04
+ARG BASE_IMAGE
 FROM ${BASE_IMAGE}
 
 LABEL maintainer="MSCCL++"