Skip to content

Commit

Permalink
Limited CUDA builds to a single CUDA version per major CUDA version
Browse files Browse the repository at this point in the history
  • Loading branch information
stemann committed Aug 24, 2022
1 parent d0e73a8 commit a81fa6b
Showing 1 changed file with 7 additions and 11 deletions.
18 changes: 7 additions & 11 deletions O/ONNXRuntime/ONNXRuntime/build_tarballs.jl
Original file line number Diff line number Diff line change
Expand Up @@ -7,7 +7,7 @@ version = v"1.10.0"

# Cf. https://onnxruntime.ai/docs/execution-providers/CUDA-ExecutionProvider.html#requirements
# Cf. https://onnxruntime.ai/docs/execution-providers/TensorRT-ExecutionProvider.html#requirements
cuda_versions = [v"11.0", v"11.1", v"11.2", v"11.3"] # No CUDA 10.2, since pre-built x86_64 ONNXRuntime-CUDA binaries are built for CUDA 11
cuda_tag = "11.3" # No CUDA 10.2, since pre-built x86_64 ONNXRuntime-CUDA binaries are built for CUDA 11
cuda_aarch64_tag = "10.2"
cudnn_version = v"8.2.4"
tensorrt_version = v"8.0.1"
Expand Down Expand Up @@ -89,16 +89,12 @@ end
platforms = supported_platforms(; exclude=platform_exclude_filter)
platforms = expand_cxxstring_abis(platforms; skip=!Sys.islinux)

cuda_platforms = Platform[]
for cuda_version in cuda_versions
cuda_tag = "$(cuda_version.major).$(cuda_version.minor)"
for p in [
Platform("x86_64", "Linux"; cuda = cuda_tag),
Platform("x86_64", "Windows"; cuda = cuda_tag)
]
push!(platforms, p)
push!(cuda_platforms, p)
end
for p in [
Platform("x86_64", "Linux"; cuda = cuda_tag),
Platform("x86_64", "Windows"; cuda = cuda_tag)
]
push!(platforms, p)
push!(cuda_platforms, p)
end
push!(platforms, Platform("aarch64", "Linux"; cuda = cuda_aarch64_tag))
push!(cuda_platforms, Platform("aarch64", "Linux"; cuda = cuda_aarch64_tag))
Expand Down

0 comments on commit a81fa6b

Please sign in to comment.