Skip to content

Commit cb968a8

Browse files
committed
update yml
1 parent fe7dd53 commit cb968a8

File tree

2 files changed

+5
-4
lines changed

2 files changed

+5
-4
lines changed

tools/ci_build/github/azure-pipelines/stages/py-gpu-packaging-stage.yml

+2-2
Original file line numberDiff line numberDiff line change
@@ -56,7 +56,7 @@ stages:
5656
PYTHON_VERSION: ${{ python_version }}
5757
EP_NAME: gpu
5858
CudaVersion: ${{ parameters.cuda_version }}
59-
EP_BUILD_FLAGS: --enable_lto --cuda_home=$(Agent.TempDirectory)\v${{ parameters.cuda_version }} --cmake_extra_defines "CMAKE_CUDA_ARCHITECTURES=75;80;90"
59+
EP_BUILD_FLAGS: --enable_lto --use_cuda --cuda_version ${{ parameters.cuda_version }} --cuda_home=$(Agent.TempDirectory)\v${{ parameters.cuda_version }} --cmake_extra_defines "CMAKE_CUDA_ARCHITECTURES=75;80;90"
6060
use_tensorrt: True
6161

6262
- ${{ if eq(parameters.enable_linux_cuda, true) }}:
@@ -80,4 +80,4 @@ stages:
8080
PYTHON_VERSION: ${{ python_version }}
8181
EP_BUILD_FLAGS: --use_dml --cmake_extra_defines CMAKE_SYSTEM_VERSION=10.0.18362.0 --enable_wcos
8282
EP_NAME: directml
83-
cmake_build_type: ${{ parameters.cmake_build_type }}
83+
cmake_build_type: ${{ parameters.cmake_build_type }}

tools/ci_build/github/azure-pipelines/stages/py-win-gpu-stage.yml

+3-2
Original file line numberDiff line numberDiff line change
@@ -33,7 +33,7 @@ parameters:
3333
- Release
3434
- RelWithDebInfo
3535
- MinSizeRel
36-
36+
3737
- name: use_tensorrt
3838
type: boolean
3939
default: false
@@ -125,7 +125,7 @@ stages:
125125
--cmake_generator "$(VSGenerator)"
126126
--enable_pybind
127127
--enable_onnx_tests
128-
--parallel 8 --use_vcpkg --use_binskim_compliant_compile_flags --update --build --msvc_toolset 14.40
128+
--parallel 8 --use_binskim_compliant_compile_flags --update --build --msvc_toolset 14.40
129129
$(TelemetryOption) ${{ parameters.BUILD_PY_PARAMETERS }} ${{ parameters.EP_BUILD_FLAGS }} ${{ variables.trt_build_flag }}
130130
workingDirectory: '$(Build.BinariesDirectory)'
131131

@@ -195,6 +195,7 @@ stages:
195195
TMPDIR: "$(Agent.TempDirectory)"
196196

197197
- powershell: |
198+
198199
python -m pip uninstall -y onnxruntime onnxruntime-gpu -qq
199200
Get-ChildItem -Path $(Build.ArtifactStagingDirectory)/*cp${{ replace(parameters.PYTHON_VERSION,'.','') }}*.whl | foreach {pip --disable-pip-version-check install --upgrade $_.fullname tabulate}
200201
mkdir -p $(Agent.TempDirectory)\ort_test_data

0 commit comments

Comments
 (0)