adds win arm64 test/build stages to v2 pipeline

.pipelines/templates/fetch-test-data-from-blob.yml

@@ -5,13 +5,63 @@ parameters:
 - name: destinationPath
   type: string
   default: '$(Build.SourcesDirectory)/test-data-shared'
+- name: scriptType
+  type: string
+  default: pscore
+  values: ['pscore', 'ps']
+- name: installAzCli
+  type: boolean
+  default: false
 
 steps:
+- ${{ if eq(parameters.installAzCli, true) }}:
+  - task: PowerShell@2
+    displayName: 'Ensure Azure CLI is installed (Windows)'
+    condition: eq(variables['Agent.OS'], 'Windows_NT')
+    inputs:
+      targetType: inline
+      pwsh: false
+      script: |
+        $ErrorActionPreference = 'Stop'
+
+        if (Get-Command az -ErrorAction SilentlyContinue) {
+          $azDir = Split-Path -Path (Get-Command az).Source -Parent
+          Write-Host "##vso[task.prependpath]$azDir"
+          az --version | Select-Object -First 1 | Write-Host
+          Write-Host 'Azure CLI already installed.'
+          exit 0
+        }
+
+        try {
+          $installerPath = Join-Path $env:TEMP 'azure-cli-installer.msi'
+          $installerUrl = 'https://aka.ms/installazurecliwindows'
+
+          Write-Host "Downloading Azure CLI installer from $installerUrl"
+          Invoke-WebRequest -Uri $installerUrl -OutFile $installerPath -UseBasicParsing
+
+          Write-Host 'Installing Azure CLI via MSI...'
+          $process = Start-Process -FilePath 'msiexec.exe' -ArgumentList "/i `"$installerPath`" /qn /norestart" -Wait -PassThru
+          if ($process.ExitCode -ne 0) {
+            throw "Azure CLI MSI install failed with exit code $($process.ExitCode)."
+          }
+        }
+        catch {
+          throw "Azure CLI install failed: $($_.Exception.Message)"
+        }
+
+        $env:PATH = [System.Environment]::GetEnvironmentVariable('Path', 'Machine') + ';' + [System.Environment]::GetEnvironmentVariable('Path', 'User')
+        $azCommand = Get-Command az -ErrorAction SilentlyContinue
+        if (-not $azCommand) {
+          throw 'Azure CLI install was attempted but az is still not on PATH.'
+        }
+        Write-Host "##vso[task.prependpath]$(Split-Path -Path $azCommand.Source -Parent)"
+        az version | Write-Host
+
 - task: AzureCLI@2
   displayName: 'Fetch test data from blob'
   inputs:
     azureSubscription: 'ortcibuild_readonly_mi2-ONNX Runtime-AIFoundryLocal'
-    scriptType: pscore
+    scriptType: ${{ parameters.scriptType }}
     scriptLocation: inlineScript
     inlineScript: |
       $ErrorActionPreference = 'Stop'
@@ -21,7 +71,8 @@ steps:
       $container = 'models'
       $prefix = 'foundrylocal/models'
 
-      if (-not (Get-Command azcopy -ErrorAction SilentlyContinue)) {
+      $azcopyCommand = Get-Command azcopy -ErrorAction SilentlyContinue
+      if (-not $azcopyCommand) {
         if ($IsMacOS) {
           Write-Host 'azcopy not found on macOS agent. Installing via Homebrew...'
           if (-not (Get-Command brew -ErrorAction SilentlyContinue)) {
@@ -30,12 +81,32 @@ steps:
 
           brew update --quiet
           brew install --quiet azcopy
+        } elseif ($env:AGENT_OS -eq 'Windows_NT' -and $env:AGENT_OSARCHITECTURE -eq 'ARM64') {
+          Write-Host 'azcopy not found on Windows agent. Installing from official zip...'
+          $zipPath = Join-Path $env:TEMP 'azcopy_windows.zip'
+          $extractRoot = Join-Path $env:TEMP 'azcopy'
+          Invoke-WebRequest -Uri 'https://aka.ms/downloadazcopy-v10-windows' -OutFile $zipPath -UseBasicParsing
+          if (Test-Path $extractRoot) {
+            Remove-Item -Path $extractRoot -Recurse -Force
+          }
+          Expand-Archive -Path $zipPath -DestinationPath $extractRoot -Force
+          $azcopyExe = Get-ChildItem -Path $extractRoot -Filter 'azcopy.exe' -Recurse | Select-Object -First 1
+          if (-not $azcopyExe) {
+            throw 'azcopy install completed but azcopy.exe was not found.'
+          }
+          $env:PATH = "$($azcopyExe.Directory.FullName);$env:PATH"
         } else {
           throw 'azcopy is not installed on this agent.'
         }
+
+        $azcopyCommand = Get-Command azcopy -ErrorAction SilentlyContinue
+      }
+
+      if (-not $azcopyCommand) {
+        throw 'azcopy is not available on PATH after installation/probe.'
       }
 
-      azcopy --version
+      & $azcopyCommand.Source --version | Write-Host
 
       New-Item -ItemType Directory -Path $destination -Force | Out-Null
       $publisherRoot = Join-Path $destination 'Microsoft'

.pipelines/v2/templates/stages-build-native.yml

@@ -57,7 +57,7 @@ stages:
           stageHeaders: true
 
   # ====================================================================
-  # Windows ARM64 — cross-compile only (no tests on x64 host)
+  # Windows ARM64 — build + test
   # ====================================================================
   - stage: cpp_build_win_arm64
     displayName: 'C++ Native: Windows ARM64'
@@ -66,8 +66,9 @@ stages:
     jobs:
     - job: build
       pool:
-        name: onnxruntime-Win-CPU-2022
+        name: onnxruntime-native-arm64-16core
         os: windows
+        hostArchitecture: arm64
       templateContext:
         inputs:
         - input: pipelineArtifact
@@ -85,8 +86,10 @@ stages:
           ortVersion: ${{ parameters.ortVersion }}
           genaiVersion: ${{ parameters.genaiVersion }}
           winmlVersion: ${{ parameters.winmlVersion }}
-          runTests: false
+          runTests: true
           stageHeaders: false
+          pythonArchitecture: arm64
+          usePwsh: false
 
   # ====================================================================
   # Linux x64 — build + test

.pipelines/v2/templates/stages-cs.yml

@@ -68,10 +68,45 @@ stages:
     - checkout: self
       clean: true
     - template: ../../templates/fetch-test-data-from-blob.yml@self
+      parameters:
+        scriptType: ps
+        installAzCli: true
+    - template: steps-test-cs.yml
+      parameters:
+        flNugetDir: '$(Pipeline.Workspace)/cpp-nuget'
+        testDataSharedDir: '$(Build.SourcesDirectory)/test-data-shared'
+        usePwsh: false
+
+- stage: cs_test_win_arm64
+  displayName: 'C# SDK: Test Windows ARM64'
+  dependsOn:
+  - cs_build
+  jobs:
+  - job: test
+    pool:
+      name: onnxruntime-native-arm64-16core
+      os: windows
+      hostArchitecture: arm64
+    templateContext:
+      inputs:
+      - input: pipelineArtifact
+        artifactName: 'version-info'
+        targetPath: '$(Pipeline.Workspace)/version-info'
+      - input: pipelineArtifact
+        artifactName: 'cpp-nuget'
+        targetPath: '$(Pipeline.Workspace)/cpp-nuget'
+    steps:
+    - checkout: self
+      clean: true
+    - template: ../../templates/fetch-test-data-from-blob.yml@self
+      parameters:
+        scriptType: ps
+        installAzCli: true
     - template: steps-test-cs.yml
       parameters:
         flNugetDir: '$(Pipeline.Workspace)/cpp-nuget'
         testDataSharedDir: '$(Build.SourcesDirectory)/test-data-shared'
+        usePwsh: false
 
 # ====================================================================
 # Test — Linux x64

.pipelines/v2/templates/stages-js.yml

@@ -13,8 +13,7 @@
 #   * cpp_build_win_x64        (the cpp-native-include artifact, which is
 #                               stage-headers-only on the win-x64 build)
 #
-# Tests run on the same platform as the build, except win-arm64 (no
-# Windows-ARM64 pool; build-only, matching the C# + Python matrices).
+# Tests run on the same platform as the build.
 
 stages:
 
@@ -69,8 +68,9 @@ stages:
   jobs:
   - job: build
     pool:
-      name: onnxruntime-Win-CPU-2022
+      name: onnxruntime-native-arm64-16core
       os: windows
+      hostArchitecture: arm64
     variables:
     - group: FoundryLocal-ESRP-Signing
     templateContext:
@@ -97,7 +97,6 @@ stages:
         nativeArtifactDir: '$(Pipeline.Workspace)/cpp-native-win-arm64'
         includeArtifactDir: '$(Pipeline.Workspace)/cpp-native-include'
         outputDir: '$(Build.ArtifactStagingDirectory)/js-prebuild'
-        targetArch: 'arm64'
         signWindows: true
 
 - stage: js_build_linux_x64
@@ -214,9 +213,9 @@ stages:
         outputDir: '$(Build.ArtifactStagingDirectory)/js-prebuild'
         signMac: true
 
-# =====================================================================
-# Test stages — same-platform only. win-arm64 is build-only.
-# =====================================================================
+# ================================================================
+# Test stages — same-platform only (including native win-arm64).
+# ================================================================
 
 - stage: js_test_win_x64
   displayName: 'JS SDK: Test Windows x64'
@@ -242,6 +241,35 @@ stages:
         prebuildArtifactDir: '$(Pipeline.Workspace)/js-prebuild-win-x64'
         testDataSharedDir: '$(Build.SourcesDirectory)/test-data-shared'
 
+- stage: js_test_win_arm64
+  displayName: 'JS SDK: Test Windows ARM64'
+  dependsOn:
+  - js_build_win_arm64
+  jobs:
+  - job: test
+    pool:
+      name: onnxruntime-native-arm64-16core
+      os: windows
+      hostArchitecture: arm64
+    templateContext:
+      inputs:
+      - input: pipelineArtifact
+        artifactName: 'js-prebuild-win-arm64'
+        targetPath: '$(Pipeline.Workspace)/js-prebuild-win-arm64'
+    steps:
+    - checkout: self
+      clean: true
+    - template: ../../templates/fetch-test-data-from-blob.yml@self
+      parameters:
+        scriptType: ps
+        installAzCli: true
+    - template: steps-test-js.yml
+      parameters:
+        rid: 'win-arm64'
+        prebuildArtifactDir: '$(Pipeline.Workspace)/js-prebuild-win-arm64'
+        testDataSharedDir: '$(Build.SourcesDirectory)/test-data-shared'
+        usePwsh: false
+
 - stage: js_test_linux_x64
   displayName: 'JS SDK: Test Linux x64'
   dependsOn:

.pipelines/v2/templates/stages-python.yml

@@ -1,8 +1,7 @@
 # Build + test stages for the sdk_v2 Python SDK (foundry-local-sdk).
 #
 # Builds win-x64 + win-arm64 + linux-x64 + linux-arm64 + osx-arm64;
-# tests win-x64 + linux-x64 + linux-arm64 + osx-arm64
-# (win-arm64 is build-only, matching the C# matrix).
+# tests win-x64 + win-arm64 + linux-x64 + linux-arm64 + osx-arm64;
 #
 # Each build stage depends on the matching native build stage from
 # stages-build-native.yml and consumes the `cpp-native-<rid>` pipeline artifact
@@ -56,8 +55,9 @@ stages:
   jobs:
   - job: build
     pool:
-      name: onnxruntime-Win-CPU-2022
+      name: onnxruntime-native-arm64-16core
       os: windows
+      hostArchitecture: arm64
     templateContext:
       inputs:
       - input: pipelineArtifact
@@ -77,7 +77,7 @@ stages:
       parameters:
         nativeArtifactDir: '$(Pipeline.Workspace)/cpp-native-win-arm64'
         rid: 'win-arm64'
-        targetArch: 'arm64'
+        pythonArchitecture: 'arm64'
         outputDir: '$(Build.ArtifactStagingDirectory)/python-sdk'
 
 - stage: python_build_linux_x64
@@ -178,9 +178,9 @@ stages:
         pythonArchitecture: 'arm64'
         outputDir: '$(Build.ArtifactStagingDirectory)/python-sdk'
 
-# =================================================================
-# Tests — win-x64, linux-x64, osx-arm64. win-arm64 is build-only.
-# =================================================================
+# ======================================================
+# Tests — win-x64, win-arm64, linux-x64, osx-arm64.
+# ======================================================
 - stage: python_test_win_x64
   displayName: 'Python SDK: Test Windows x64'
   dependsOn:
@@ -206,6 +206,35 @@ stages:
         wheelDir: '$(Pipeline.Workspace)/python-sdk-win-x64'
         testDataSharedDir: '$(Build.SourcesDirectory)/test-data-shared'
 
+- stage: python_test_win_arm64
+  displayName: 'Python SDK: Test Windows ARM64'
+  dependsOn:
+  - python_build_win_arm64
+  jobs:
+  - job: test
+    pool:
+      name: onnxruntime-native-arm64-16core
+      os: windows
+      hostArchitecture: arm64
+    templateContext:
+      inputs:
+      - input: pipelineArtifact
+        artifactName: 'python-sdk-win-arm64'
+        targetPath: '$(Pipeline.Workspace)/python-sdk-win-arm64'
+    steps:
+    - checkout: self
+      clean: true
+    - template: ../../templates/fetch-test-data-from-blob.yml@self
+      parameters:
+        scriptType: ps
+        installAzCli: true
+    - template: steps-test-python.yml
+      parameters:
+        wheelDir: '$(Pipeline.Workspace)/python-sdk-win-arm64'
+        testDataSharedDir: '$(Build.SourcesDirectory)/test-data-shared'
+        pythonArchitecture: 'arm64'
+        usePwsh: false
+
 - stage: python_test_linux_x64
   displayName: 'Python SDK: Test Linux x64'
   dependsOn: