Skip to content

Commit 6146680

Browse files
Update adapt.jl
1 parent e26109d commit 6146680

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

src/adapt.jl

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -17,9 +17,9 @@ function adapt_structure(to,
1717
end
1818

1919
# Allow DAE adaptation for GPU kernels
20-
function adapt_structure(to, f::SciMLBase.ODEFunction{iip}) where {iip}
20+
function adapt_structure(to, f::ODEFunction{iip}) where {iip}
2121
# For GPU kernels, we now support DAEs with mass matrices and initialization
22-
SciMLBase.ODEFunction{iip, SciMLBase.FullSpecialize}(
22+
ODEFunction{iip, FullSpecialize}(
2323
f.f,
2424
jac = f.jac,
2525
mass_matrix = f.mass_matrix,
@@ -28,8 +28,8 @@ function adapt_structure(to, f::SciMLBase.ODEFunction{iip}) where {iip}
2828
end
2929

3030
# Adapt OverrideInitData for GPU compatibility
31-
function adapt_structure(to, f::SciMLBase.OverrideInitData)
32-
SciMLBase.OverrideInitData(
31+
function adapt_structure(to, f::OverrideInitData)
32+
OverrideInitData(
3333
adapt(to, f.initializeprob), # Also adapt initializeprob
3434
f.update_initializeprob!,
3535
f.initializeprobmap,

0 commit comments

Comments
 (0)