Skip to content

Commit

Permalink
Versioning/compat
Browse files Browse the repository at this point in the history
  • Loading branch information
stemann committed Feb 7, 2022
1 parent 6f92c80 commit 93cf3f8
Showing 1 changed file with 10 additions and 3 deletions.
13 changes: 10 additions & 3 deletions O/onnxruntime/onnxruntime_providers_cuda/build_tarballs.jl
Original file line number Diff line number Diff line change
Expand Up @@ -5,6 +5,13 @@ using BinaryBuilder, Pkg
name = "onnxruntime_providers_cuda"
version = v"1.10.0"

# Cf. https://onnxruntime.ai/docs/execution-providers/CUDA-ExecutionProvider.html#requirements
cuda_version = v"11.4"
cudnn_version = v"8.2.4"

cuda_compat = cuda_version.major
cudnn_compat = cudnn_version.major

# Collection of sources required to complete build
sources = [
ArchiveSource("https://github.com/microsoft/onnxruntime/releases/download/v$version/onnxruntime-linux-x64-gpu-$version.tgz", "bc880ba8a572acf79d50dcd35ba6dd8e5fb708d03883959ef60efbc15f5cdcb6"),
Expand All @@ -30,8 +37,8 @@ install_license $dist_name*/LICENSE
# These are the platforms we will build for by default, unless further
# platforms are passed in on the command line
platforms = [
Platform("x86_64", "Linux"; cuda = "11.4"),
Platform("x86_64", "Windows"; cuda = "11.4")
Platform("x86_64", "Linux"; cuda = string(cuda_version)),
Platform("x86_64", "Windows"; cuda = string(cuda_version))
]
platforms = expand_cxxstring_abis(platforms; skip=!Sys.islinux)

Expand All @@ -44,7 +51,7 @@ products = [

# Dependencies that must be installed before this package can be built
dependencies = [
Dependency("CUDNN_jll", v"8.2.0"),
Dependency("CUDNN_jll", v"8.2.2"; compat = cudnn_compat), # TODO build_version should actually be cudnn_version (v"8.2.4")
Dependency("Zlib_jll")
]

Expand Down

0 comments on commit 93cf3f8

Please sign in to comment.