diff --git a/.github/workflows/build-wheels-windows.yml b/.github/workflows/build-wheels-windows.yml index 818d9f78b08..a269aea2604 100644 --- a/.github/workflows/build-wheels-windows.yml +++ b/.github/workflows/build-wheels-windows.yml @@ -25,6 +25,7 @@ jobs: os: windows test-infra-repository: pytorch/test-infra test-infra-ref: main + with-xpu: enable build: needs: generate-matrix strategy: diff --git a/packaging/windows/internal/vc_env_helper.bat b/packaging/windows/internal/vc_env_helper.bat index d3484a66e9f..699876beb8a 100644 --- a/packaging/windows/internal/vc_env_helper.bat +++ b/packaging/windows/internal/vc_env_helper.bat @@ -28,6 +28,8 @@ if "%VSDEVCMD_ARGS%" == "" ( @echo on +if "%CU_VERSION%" == "xpu" call "C:\Program Files (x86)\Intel\oneAPI\setvars.bat" + set DISTUTILS_USE_SDK=1 set args=%1