Skip to content

Commit 9d27905

Browse files
committed
Using CUDA platform selection
Also, * Replaced lazy with eager artifacts.
1 parent e40b2dd commit 9d27905

File tree

1 file changed

+5
-4
lines changed

1 file changed

+5
-4
lines changed

O/ONNXRuntime/build_tarballs.jl

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -2,14 +2,15 @@
22
# `julia build_tarballs.jl --help` to see a usage message.
33
using BinaryBuilder, Pkg
44

5+
include(joinpath(@__DIR__, "..", "..", "platforms", "cuda.jl"))
6+
57
name = "ONNXRuntime"
68
version = v"1.10.0"
79

810
# Cf. https://onnxruntime.ai/docs/execution-providers/CUDA-ExecutionProvider.html#requirements
911
# Cf. https://onnxruntime.ai/docs/execution-providers/TensorRT-ExecutionProvider.html#requirements
1012
cuda_versions = [
11-
# No CUDA 10.2, since pre-built x86_64 CUDA binaries are built for CUDA 11
12-
v"11.3", # Using 11.3, and not 11.4, to be compatible with TensorRT (JLL) v8.0.1 (which includes aarch64 support)
13+
v"11.3.1", # Using 11.3, and not 11.4, to be compatible with TensorRT (JLL) v8.0.1 (which includes aarch64 support)
1314
]
1415
cuda_aarch64_tag = "10.2"
1516
cudnn_version = v"8.2.4"
@@ -134,12 +135,12 @@ dependencies = [
134135
compat = tensorrt_compat,
135136
platforms = cuda_platforms),
136137
Dependency("Zlib_jll"; platforms = cuda_platforms),
137-
BuildDependency(PackageSpec("CUDA_full_jll", v"11.3.1"); platforms = cuda_platforms),
138+
BuildDependency(PackageSpec("CUDA_full_jll", first(cuda_versions)); platforms = cuda_platforms),
138139
HostBuildDependency(PackageSpec("protoc_jll", v"3.16.1"))
139140
]
140141

141142
# Build the tarballs, and possibly a `build.jl` as well.
142143
build_tarballs(ARGS, name, version, sources, script, platforms, products, dependencies;
143-
lazy_artifacts = true,
144+
augment_platform_block = CUDA.augment,
144145
julia_compat = "1.6",
145146
preferred_gcc_version = v"8")

0 commit comments

Comments
 (0)