[On-Device Training] Expose Parameters through the Training API (#17364) #7
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
name: Windows_CI | |
on: | |
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 | |
jobs: | |
Windows-CUDA-12: | |
runs-on: ["self-hosted", "1ES.Pool=onnxruntime-github-vs2022-mms"] | |
steps: | |
- uses: actions/checkout@v4 | |
with: | |
submodules: false | |
- uses: actions/setup-python@v4 | |
with: | |
python-version: '3.11.x' | |
architecture: 'x64' | |
- uses: actions/setup-node@v3 | |
with: | |
node-version: 18 | |
- name: Download cuda | |
run: azcopy.exe cp --recursive "https://lotusscus.blob.core.windows.net/models/cuda_sdk/v12.2" cuda_sdk | |
- name: Delete build folder | |
run: | | |
if (Test-Path D:\b) { Remove-Item -Recurse -Force D:\b } | |
&tools\ci_build\github\windows\install_third_party_deps.ps1 -cpu_arch x64 -install_prefix D:\b\Debug\installed -build_config Debug | |
# The build machine doesn't have a GPU. So the value of CMAKE_CUDA_ARCHITECTURES doesn't matter. | |
- name: Build code | |
run: python tools\ci_build\build.py --windows_sdk_version 10.0.22621.0 --enable_training --build_java --config Debug --build_dir D:\b --skip_submodule_sync --build_csharp --update --build --parallel --cmake_generator "Visual Studio 17 2022" --build_shared_lib --enable_pybind --use_cuda --cuda_home=${{ github.workspace }}\cuda_sdk\v12.2 --enable_cuda_profiling --cmake_extra_defines CMAKE_CUDA_ARCHITECTURES=75 | |
Onnxruntime-TVM: | |
runs-on: windows-latest | |
steps: | |
- uses: actions/checkout@v4 | |
with: | |
submodules: true | |
- uses: actions/setup-python@v4 | |
with: | |
python-version: '3.8.x' | |
architecture: 'x64' | |
- uses: conda-incubator/setup-miniconda@v2 | |
with: | |
activate-environment: "" | |
- name: 'Install LLVM-Dev' | |
shell: pwsh | |
run: | | |
conda install llvmdev=12.0.0 | |
conda info | |
conda list | |
- name: 'Add LLVM-Dev binaries to the PATH' | |
run: | | |
echo "C:/Miniconda/Library/bin" >> $GITHUB_PATH | |
- name: 'Setup TVM EP Python requirements' | |
run: | | |
python3 -m pip install -r ${{ github.workspace }}/tools/ci_build/github/linux/tvm/requirements.txt | |
- name: 'rm gtest in conda' | |
shell: pwsh | |
run: | | |
Remove-Item 'C:\Miniconda\Library\lib\cmake\gtest' -Recurse -Force | |
Remove-Item 'C:\Miniconda\Library\lib\gmock.lib' -Force | |
Remove-Item 'C:\Miniconda\Library\lib\gmock_main.lib' -Force | |
Remove-Item 'C:\Miniconda\Library\lib\gtest.lib' -Force | |
Remove-Item 'C:\Miniconda\Library\lib\gtest_main.lib' -Force | |
- name: 'Build and Test' | |
run: | | |
python3 ${{ github.workspace }}/tools/ci_build/build.py --build_dir build --config Release --skip_submodule_sync --parallel --enable_pybind --disable_contrib_ops --disable_ml_ops --skip_onnx_tests --use_tvm |