From eaf21ecb062b60e44ad1cb5490ac9fa3d0f5d5a0 Mon Sep 17 00:00:00 2001
From: Aditya Patwardhan <adityap@microsoft.com>
Date: Mon, 6 May 2024 16:23:08 -0700
Subject: [PATCH] Use PSScriptRoot to find path to Wix module (#21611)

---
 build.psm1 | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/build.psm1 b/build.psm1
index 1b169d62c99..38dd4acc3db 100644
--- a/build.psm1
+++ b/build.psm1
@@ -2295,6 +2295,11 @@ function Start-PSBootstrap {
                 $psInstallFile = [System.IO.Path]::Combine($PSScriptRoot, "tools", "install-powershell.ps1")
                 & $psInstallFile -AddToPath
             }
+            if ($Package) {
+                Import-Module "$PSScriptRoot\tools\wix\wix.psm1"
+                $isArm64 = '$(Runtime)' -eq 'arm64'
+                Install-Wix -arm64:$isArm64
+            }
         }
     } finally {
         Pop-Location