fix detiming

Author: ChrisRackauckas

src/systems/diffeqs/diffeqsystem.jl

@@ -144,7 +144,9 @@ function calculate_tgrad(sys::ODESystem)
   rhs = [detime_dvs(eq.rhs) for eq ∈ sys.eqs]
   iv = sys.iv()
   notime_tgrad = [expand_derivatives(ModelingToolkit.Differential(iv)(r)) for r in rhs]
+  @show notime_tgrad
   tgrad = retime_dvs.(notime_tgrad,(sys.dvs,),iv)
+  @show tgrad
   sys.tgrad[] = tgrad
   return tgrad
 end

src/utils.jl

@@ -37,7 +37,7 @@ end
 
 function detime_dvs(op::Operation)
   if op.op isa Variable
-    Operation(Variable(op.op.name),Expression[])
+    Operation(Variable(op.op.name,known=op.op.known),Expression[])
   else
     Operation(op.op,detime_dvs.(op.args))
   end

test/system_construction.jl

@@ -74,6 +74,18 @@ sol = Sfw_t \ @SArray ones(3)
 @test sol isa SArray
 @test sol ≈ -(I/0.2 - J)\ones(3)
 
+eqs = [D(x) ~ σ*(y-x),
+       D(y) ~ x*(ρ-z)-y*t,
+       D(z) ~ x*y - β*z]
+de = ODESystem(eqs)
+ModelingToolkit.calculate_tgrad(de)
+
+tgrad_oop, tgrad_iip = eval.(ModelingToolkit.generate_tgrad(de))
+@test tgrad_oop(u,p,t) == [0.0,-u[2],0.0]
+du = zeros(3)
+tgrad_iip(du,u,p,t)
+@test du == [0.0,-u[2],0.0]
+
 @testset "time-varying parameters" begin
     @parameters σ′(t-1)
     eqs = [D(x) ~ σ′*(y-x),