Skip to content

Commit dadbda3

Browse files
authored
Update modeler instantiation for CUDA backend
1 parent 9305d0f commit dadbda3

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

test/suite/builders/test_options_forwarding.jl

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -54,7 +54,7 @@ function test_options_forwarding()
5454
if is_cuda_on()
5555
Test.@testset "backend (CUDA)" begin
5656
Test.@test begin
57-
modeler = OptimalControl.Exa(backend=CUDA.CUDABackend())
57+
modeler = OptimalControl.Exa{OptimalControl.GPU}(backend=CUDA.CUDABackend())
5858
nlp = OptimalControl.nlp_model(docp, normalized_init, modeler)
5959
# With CUDA backend, x0 should be CUDA array
6060
nlp.meta.x0 isa CUDA.CuArray

0 commit comments

Comments
 (0)