-
Notifications
You must be signed in to change notification settings - Fork 100
105 lines (88 loc) · 3.44 KB
/
win-directml-x64-build.yml
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
name: "Windows DirectML x64 Build"
on:
workflow_dispatch:
push:
branches:
- main
- rel-*
pull_request:
concurrency:
group: ${{ github.workflow }}-${{ github.head_ref || github.run_id }}
cancel-in-progress: true
env:
AZCOPY_AUTO_LOGIN_TYPE: MSI
AZCOPY_MSI_CLIENT_ID: 63b63039-6328-442f-954b-5a64d124e5b4
ort_dir: "Microsoft.ML.OnnxRuntime.DirectML.1.17.3"
ort_zip: "Microsoft.ML.OnnxRuntime.DirectML.1.17.3.zip"
# TODO: Update with nightly ORT-DML build
ort_url: "https://github.com/microsoft/onnxruntime/releases/download/v1.17.3/Microsoft.ML.OnnxRuntime.DirectML.1.17.3.zip"
dml_dir: "Microsoft.AI.DirectML.1.15.1"
dml_zip: "Microsoft.AI.DirectML.1.15.1.zip"
dml_url: "https://www.nuget.org/api/v2/package/Microsoft.AI.DirectML/1.15.1"
d3d12_dir: "Microsoft.Direct3D.D3D12.1.614.0"
d3d12_zip: "Microsoft.Direct3D.D3D12.1.614.0.zip"
d3d12_url: "https://www.nuget.org/api/v2/package/Microsoft.Direct3D.D3D12/1.614.0"
binaryDir: 'build/directml/win-x64'
jobs:
windows-directml-x64-build:
runs-on: [ "self-hosted", "1ES.Pool=onnxruntime-genai-Win2022-GPU-A10" ]
steps:
- name: Checkout OnnxRuntime GenAI repo
uses: actions/checkout@v4
with:
submodules: true
- uses: actions/setup-python@v5
with:
python-version: '3.11.x'
architecture: 'x64'
- name: Download OnnxRuntime
run: |
Invoke-WebRequest -Uri $env:ort_url -OutFile $env:ort_zip
- name: Download DirectML
run: |
Invoke-WebRequest -Uri $env:dml_url -OutFile $env:dml_zip
- name: Download the D3D12 Agility SDK
run: |
Invoke-WebRequest -Uri $env:d3d12_url -OutFile $env:d3d12_zip
- name: Unzip OnnxRuntime
run: |
Expand-Archive $env:ort_zip -DestinationPath $env:ort_dir
Remove-Item -Path $env:ort_zip
- name: Unzip DirectML
run: |
Expand-Archive $env:dml_zip -DestinationPath $env:dml_dir
Remove-Item -Path $env:dml_zip
- name: Unzip the D3D12 Agility SDK
run: |
Expand-Archive $env:d3d12_zip -DestinationPath $env:d3d12_dir
Remove-Item -Path $env:d3d12_zip
- name: Move the files to the ort directory
run: |
mkdir ort/lib
mkdir ort/include
mv $env:ort_dir\runtimes\win-x64\native\onnxruntime.dll ort\lib
mv $env:ort_dir\runtimes\win-x64\native\onnxruntime.lib ort\lib
mv $env:dml_dir\bin\x64-win\DirectML.dll ort\lib
mv $env:d3d12_dir\build\native\bin\x64\D3D12Core.dll ort\lib
mv $env:ort_dir\build\native\include\dml_provider_factory.h ort\include
mv $env:ort_dir\build\native\include\onnxruntime_c_api.h ort\include
mv $env:dml_dir\include\DirectML.h ort\include
- name: Configure CMake
run: |
cmake --preset windows_x64_directml_release -DTEST_PHI2=False
- name: Build with CMake
run: |
cmake --build --preset windows_x64_directml_release --parallel
- name: Install the Python Wheel and Test Dependencies
run: |
python -m pip install -r test\python\requirements.txt
python -m pip install -r test\python\requirements-directml.txt
python -m pip install (Get-ChildItem ("$env:binaryDir\wheel\*.whl"))
- name: Verify Build Artifacts
if: always()
continue-on-error: true
run: |
Get-ChildItem -Path $env:GITHUB_WORKSPACE\$env:binaryDir -Recurse
- name: Run tests
run: |
& .\$env:binaryDir\test\Release\unit_tests.exe