Update

[ghstack-poisoned]

Author: GregoryComer

.ci/scripts/test_model.ps1

@@ -9,6 +9,39 @@ Set-PSDebug -Trace 1
 $ErrorActionPreference = 'Stop'
 $PSNativeCommandUseErrorActionPreference = $true
 
+function ExportModel-Portable {
+    param (
+        [string]$model_name,
+        [bool]$strict
+    )
+
+    $exportParams = "--model_name", "$modelName"
+    if ($strict) {
+        $exportParams += "--strict"
+    }
+    python -m examples.portable.scripts.export @exportParams
+    if ($LASTEXITCODE -ne 0) {
+        Write-Host "Model export failed. Exit code: $LASTEXITCODE."
+        exit $LASTEXITCODE
+    }
+
+    "$modelName.pte"
+}
+
+function ExportModel-Xnnpack {
+    param (
+        [string]$model_name
+    )
+
+    python -m examples.xnnpack.aot_compiler --model_name="${MODEL_NAME}" --delegate
+    if ($LASTEXITCODE -ne 0) {
+        Write-Host "Model export failed. Exit code: $LASTEXITCODE."
+        exit $LASTEXITCODE
+    }
+
+    "$($modelName)_xnnpack_fp32.pte"
+}
+
 .ci/scripts/setup-windows.ps1
 
 # Build the runner
@@ -27,19 +60,22 @@ $executorBinaryPath = Join-Path -Path $buildDir -ChildPath "Release\executor_run
 Pop-Location
 
 # Export the model
-$exportParams = "--model_name", "$modelName"
-if ($strict) {
-    $exportParams += "--strict"
-}
-python -m examples.portable.scripts.export @exportParams
-if ($LASTEXITCODE -ne 0) {
-    Write-Host "Model export failed. Exit code: $LASTEXITCODE."
-    exit $LASTEXITCODE
+switch ($backend) {
+    "portable" {
+        $model_path = ExportModel-Portable -model_name $modelName -strict $strict
+    }
+    "xnnpack" {
+        $model_path = ExportModel-Xnnpack -model_name $modelName
+    }
+    default {
+        Write-Host "Unknown backend $backend."
+        exit 1
+    }
 }
 
 # Run the runner
-& "$executorBinaryPath" --model_path="$modelName.pte"
+& "$executorBinaryPath" --model_path="$model_path"
 if ($LASTEXITCODE -ne 0) {
     Write-Host "Model execution failed. Exit code: $LASTEXITCODE."
     exit $LASTEXITCODE
-}
+}