Fix container workflow.

[ci skip]

Author: maleadt

.github/workflows/Container.yml

@@ -27,18 +27,18 @@ jobs:
 
     strategy:
       matrix:
-        julia: ["1.10", "1.11"]
-        cuda: ["11.8", "12.6"]
+        julia: ["1.11", "1.12"]
+        cuda: ["11.8", "12.9", "13.2"]
         platform: ["linux/amd64"]
         os: ["ubuntu-24.04"]
         include:
-          - julia: "1.11"
-            cuda: "12.6"
+          - julia: "1.12"
+            cuda: "13.2"
             platform: "linux/amd64"
             os: "ubuntu-24.04"
             default: true
-          - julia: "1.11"
-            cuda: "12.6"
+          - julia: "1.12"
+            cuda: "13.2"
             platform: "linux/arm64"
             os: "ubuntu-24.04-arm"
             default: true
@@ -48,12 +48,8 @@ jobs:
     steps:
       - name: Check out the repo
         uses: actions/checkout@v6
-
-      - name: Check out the package
-        uses: actions/checkout@v6
         with:
-          ref: ${{ inputs.ref || github.ref_name }}
-          path: package
+          ref: ${{ inputs.tag || github.ref_name }}
 
       - name: Get package spec
         id: pkg
@@ -69,7 +65,7 @@ jobs:
             echo "name=dev" >> $GITHUB_OUTPUT
           fi
 
-          VERSION=$(grep "^version = " package/Project.toml | cut -d'"' -f2)
+          VERSION=$(grep "^version = " Project.toml | cut -d'"' -f2)
           echo "version=$VERSION" >> $GITHUB_OUTPUT
 
       - name: Get CUDA major version
@@ -115,12 +111,12 @@ jobs:
         with:
           context: .
           push: true
-          provenance: false # the build fetches the repo again, so provenance tracking is not useful
+          provenance: false
           platforms: ${{ matrix.platform }}
           tags: ${{ steps.meta.outputs.tags }}
           labels: ${{ steps.meta.outputs.labels }}
           build-args: |
             JULIA_VERSION=${{ matrix.julia }}
             CUDA_VERSION=${{ matrix.cuda }}
-            PACKAGE_SPEC=CUDA#${{ steps.pkg.outputs.ref }}
+            PACKAGE_REF=${{ steps.pkg.outputs.ref }}
             JULIA_CPU_TARGET=${{ steps.cpu_target.outputs.target }}

Dockerfile

@@ -15,7 +15,7 @@ ENV JULIA_CPU_TARGET=${JULIA_CPU_TARGET}
 
 ARG CUDA_VERSION=12.6
 
-ARG PACKAGE_SPEC=CUDA
+ARG PACKAGE_REF=master
 
 LABEL org.opencontainers.image.authors="Tim Besard <tim.besard@gmail.com>" \
       org.opencontainers.image.description="A CUDA.jl container with CUDA ${CUDA_VERSION} and Julia ${JULIA_VERSION}" \
@@ -54,8 +54,17 @@ RUN julia -e '#= make bundled depot non-writable (JuliaLang/Pkg.jl#4120) =# \
     find -exec touch -h -d "@0" {} + && \
     touch -h -d "@0" /usr/local/share
 
-# install CUDA.jl itself
-RUN julia -e 'using Pkg; pkg"add '${PACKAGE_SPEC}'"; \
+# install CUDA.jl and all subpackages
+ARG PACKAGE_REF
+RUN julia -e 'using Pkg; \
+              url = "https://github.com/JuliaGPU/CUDA.jl.git"; \
+              rev = ENV["PACKAGE_REF"]; \
+              Pkg.add([PackageSpec(; url, rev, subdir) for subdir in \
+                       ["CUDACore", \
+                        "lib/cublas", "lib/cusparse", "lib/cusolver", \
+                        "lib/cufft", "lib/curand", \
+                        "lib/cudnn", "lib/cutensor", "lib/cutensornet", "lib/custatevec", \
+                        ""]]); \
               using CUDA; CUDA.precompile_runtime()' && \
     #= remove useless stuff =# \
     cd /usr/local/share/julia && \