Skip to content

Commit

Permalink
Added comment wrt. CUDA platform expansion wrt. C++ string ABI
Browse files Browse the repository at this point in the history
  • Loading branch information
stemann committed Sep 10, 2022
1 parent 92fe008 commit be022d6
Showing 1 changed file with 4 additions and 2 deletions.
6 changes: 4 additions & 2 deletions O/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_tag = "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 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 @@ -87,7 +87,6 @@ function platform_exclude_filter(p::Platform)
Sys.isfreebsd(p)
end
platforms = supported_platforms(; exclude=platform_exclude_filter)
platforms = expand_cxxstring_abis(platforms; skip=!Sys.islinux)

cuda_platforms = Platform[]
for p in [
Expand All @@ -100,6 +99,9 @@ end
push!(platforms, Platform("aarch64", "Linux"; cuda = cuda_aarch64_tag))
push!(cuda_platforms, Platform("aarch64", "Linux"; cuda = cuda_aarch64_tag))

platforms = expand_cxxstring_abis(platforms; skip=!Sys.islinux)
# Not expanding cuda_platforms wrt. cxxstring ABIs, since pre-built x86_64 CUDA binaries are built for cxx03 (and the tensorrt provider for both x86_64 and aarch64 is built for cxx11)

# The products that we will ensure are always built
products = [
LibraryProduct(["libonnxruntime", "onnxruntime"], :libonnxruntime)
Expand Down

0 comments on commit be022d6

Please sign in to comment.