Attempt to get CUDA available for build

JuliaPackaging · Feb 22, 2022 · d241894 · d241894
1 parent c7dfa37
commit d241894
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/T/Torch/TorchCUDA/build_tarballs.jl b/T/Torch/TorchCUDA/build_tarballs.jl
@@ -58,7 +58,9 @@ products = [
     LibraryProduct("libtorch", :libtorch, dont_dlopen = true),
 ]
 
-dependencies = [Dependency(PackageSpec(name="CUDA_loader_jll"))]
+dependencies = [
+    Dependency("CUDA_loader_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"]
 for cuda_version in cuda_versions
@@ -72,8 +74,9 @@ 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, [BuildDependency(PackageSpec("CUDA_jll", Base.UUID("e9e359dc-d701-5aa8-82ae-09bbf812ea83"), v"$(cuda_tag)"))])
         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