Skip to content

Commit

Permalink
Attempt to get proper, recent version of CUDA_full_jll for each CUDA …
Browse files Browse the repository at this point in the history
…version
  • Loading branch information
stemann committed Mar 5, 2022
1 parent dff04c0 commit 3b803f1
Showing 1 changed file with 16 additions and 2 deletions.
18 changes: 16 additions & 2 deletions T/Torch/TorchCUDA/build_tarballs.jl
Original file line number Diff line number Diff line change
Expand Up @@ -57,11 +57,22 @@ products = [
]

dependencies = [
BuildDependency("CUDA_full_jll"),
Dependency("CUDNN_jll")
]

cuda_versions = [v"10.2", v"11.0", v"11.1", v"11.2", v"11.3", v"11.4", v"11.5", v"11.6"]

cuda_full_version = Dict{VersionNumber,VersionNumber}(
v"10.2" => v"10.2.89",
v"11.0" => v"11.0.3",
v"11.1" => v"11.1.1",
v"11.2" => v"11.2.2",
v"11.3" => v"11.3.1",
v"11.4" => v"11.4.2",
v"11.5" => v"11.5.1",
v"11.6" => v"11.6.0"
)

for cuda_version in cuda_versions
cuda_tag = "$(cuda_version.major).$(cuda_version.minor)"
if cuda_version.major == 10
Expand All @@ -73,8 +84,11 @@ for cuda_version in cuda_versions
for platform in platforms
augmented_platform = Platform(arch(platform), os(platform); cuda=cuda_tag)
should_build_platform(triplet(augmented_platform)) || continue
platform_dependencies = vcat(dependencies, [
HostBuildDependency(PackageSpec("CUDA_full_jll", Base.UUID("4f82f1eb-248c-5f56-a42e-99106d144614"), cuda_full_version[cuda_version]))
])
build_tarballs(ARGS, name, version, sources, script, [augmented_platform],
products, dependencies; lazy_artifacts=true,
products, platform_dependencies; lazy_artifacts=true,
preferred_gcc_version = v"7.1.0")
end
end

0 comments on commit 3b803f1

Please sign in to comment.