Skip to content

Commit 434511e

Browse files
Pin CUDA runtime for Volta GPU tests
Co-Authored-By: Chris Rackauckas <accounts@chrisrackauckas.com>
1 parent 0f295fe commit 434511e

3 files changed

Lines changed: 10 additions & 3 deletions

File tree

LocalPreferences.toml

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,2 @@
1+
[CUDA_Runtime_jll]
2+
version = "12.9"

Project.toml

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -29,6 +29,7 @@ CommonSolve = "0.2"
2929
ConcreteStructs = "0.2"
3030
# cuDNN 9.11 dropped Volta support, but the GPU runner pool includes V100s.
3131
CUDNN_jll = "=9.10.0"
32+
CUDA_Runtime_jll = "0.19"
3233
DiffEqBase = "6.176.0, 7.0"
3334
Documenter = "1"
3435
FastClosures = "0.3"
@@ -63,6 +64,7 @@ julia = "1.10"
6364
[extras]
6465
Aqua = "4c88cf16-eb10-579e-8560-4a9242c79595"
6566
CUDNN_jll = "62b44479-cb7b-5706-934f-f13b2eb2e645"
67+
CUDA_Runtime_jll = "76a88914-d11a-5bdc-97e0-2f5a05c973a2"
6668
Documenter = "e30172f5-a6a5-5a46-863b-614d45cd2de4"
6769
ForwardDiff = "f6369f11-7733-5829-9624-2563aa707210"
6870
Functors = "d9f16b24-f501-4c13-a1f2-28368ffc5196"
@@ -83,4 +85,4 @@ Test = "8dfed614-e22c-5e08-85e1-65c5234f0b40"
8385
Zygote = "e88e6eb3-aa80-5325-afca-941959d7151f"
8486

8587
[targets]
86-
test = ["Aqua", "CUDNN_jll", "Documenter", "ForwardDiff", "Functors", "GPUArraysCore", "InteractiveUtils", "LuxCUDA", "LuxTestUtils", "MLDataDevices", "NLsolve", "NonlinearSolve", "OrdinaryDiffEq", "OrdinaryDiffEqAdamsBashforthMoulton", "SafeTestsets", "SciMLSensitivity", "SciMLTesting", "StableRNGs", "Test", "Zygote"]
88+
test = ["Aqua", "CUDNN_jll", "CUDA_Runtime_jll", "Documenter", "ForwardDiff", "Functors", "GPUArraysCore", "InteractiveUtils", "LuxCUDA", "LuxTestUtils", "MLDataDevices", "NLsolve", "NonlinearSolve", "OrdinaryDiffEq", "OrdinaryDiffEqAdamsBashforthMoulton", "SafeTestsets", "SciMLSensitivity", "SciMLTesting", "StableRNGs", "Test", "Zygote"]

test/shared_testsetup.jl

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -20,15 +20,18 @@ function cuda_testing()
2020
MLDataDevices.functional(CUDADevice)
2121
end
2222

23+
if BACKEND_GROUP == "all" || BACKEND_GROUP == "cuda"
24+
@test Base.pkgversion(CUDNN_jll) < v"9.11"
25+
@test cuda_testing()
26+
end
27+
2328
const MODES = begin
2429
modes = []
2530
cpu_testing() && push!(modes, ("cpu", Array, CPUDevice(), false))
2631
cuda_testing() && push!(modes, ("cuda", CuArray, CUDADevice(), true))
2732
modes
2833
end
2934

30-
cuda_testing() && @test Base.pkgversion(CUDNN_jll) < v"9.11"
31-
3235
is_finite_gradient(x::AbstractArray) = all(isfinite, x)
3336
is_finite_gradient(::Nothing) = true
3437
is_finite_gradient(gs) = all(is_finite_gradient, fleaves(gs))

0 commit comments

Comments
 (0)