Skip to content

Commit 5331b06

Browse files
authored
Update GPU modeler and solver to use GPU types
1 parent d0a330f commit 5331b06

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

test/suite/solve/test_canonical.jl

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -177,10 +177,10 @@ function test_canonical()
177177
# GPU tests (only if CUDA is available)
178178
# ----------------------------------------------------------------
179179
if is_cuda_on()
180-
gpu_modeler = ("Exa/GPU", OptimalControl.Exa(backend=CUDA.CUDABackend()))
180+
gpu_modeler = ("Exa/GPU", OptimalControl.Exa{GPU}(backend=CUDA.CUDABackend()))
181181
gpu_solver = (
182182
"MadNLP/GPU",
183-
OptimalControl.MadNLP(
183+
OptimalControl.MadNLP{GPU}(
184184
print_level=MadNLP.ERROR, linear_solver=MadNLPGPU.CUDSSSolver
185185
),
186186
)

0 commit comments

Comments
 (0)