fixes SDE

Author: shashi

src/build_function.jl

@@ -224,7 +224,6 @@ function _build_function(target::JuliaTarget, rhss::AbstractArray, args...;
                          skipzeros = outputidxs===nothing,
 						 fillzeros = skipzeros && !(typeof(rhss)<:SparseMatrixCSC),
 						 parallel=SerialForm(), kwargs...)
-    conv = conv ∘ rm_calls_with_iv
 	if multithread isa Bool
 		@warn("multithraded is deprecated for the parallel argument. See the documentation.")
 		parallel = multithread ? MultithreadedForm() : SerialForm()
@@ -524,6 +523,7 @@ function _build_function(target::CTarget, eqs::Array{<:Equation}, args...;
                          fname = :diffeqf,
 						 lhsname=:du,rhsnames=[Symbol("RHS$i") for i in 1:length(args)],
 						 libpath=tempname(),compiler=:gcc)
+
     differential_equation = string(join([numbered_expr(eq,args...,lhsname=lhsname,
                                   rhsnames=rhsnames,offset=-1) for
                                   (i, eq) ∈ enumerate(eqs)],";\n  "),";")

src/systems/diffeqs/abstractodesystem.jl

@@ -38,11 +38,10 @@ struct ODEToExpr
     states::Vector
 end
 ODEToExpr(@nospecialize(sys)) = ODEToExpr(sys,states(sys))
-function (f::ODEToExpr)(O::Operation)
+(f::ODEToExpr)(O::Num) = f(value(O))
+function (f::ODEToExpr)(O::Term)
     if isa(O.op, Sym)
-        isequal(O.op, f.sys.iv) && return O.op.name  # independent variable
-        O.op ∈ f.states         && return O.op.name  # dependent variables
-        isempty(O.args)         && return O.op.name  # 0-ary parameters
+        any(isequal(O), f.states) && return O.op.name  # dependent variables
         return build_expr(:call, Any[O.op.name; f.(O.args)])
     end
     return build_expr(:call, Any[Symbol(O.op); f.(O.args)])

src/systems/diffeqs/modelingtoolkitize.jl

@@ -8,7 +8,7 @@ function modelingtoolkitize(prob::DiffEqBase.ODEProblem)
                             return (prob.f.sys, prob.f.sys.states, prob.f.sys.ps)
     @parameters t
     var(x, i) = Sym{FnType{Tuple{symtype(t)}, Number}}(nameof(Variable(:x, i)))
-    vars = reshape([var(:x, i)(t) for i in eachindex(prob.u0)],size(prob.u0))
+    vars = reshape([var(:x, i)(value(t)) for i in eachindex(prob.u0)],size(prob.u0))
     params = prob.p isa DiffEqBase.NullParameters ? [] :
              reshape([Variable(:α,i) for i in eachindex(prob.p)],size(prob.p))
     @derivatives D'~t
@@ -40,7 +40,7 @@ function modelingtoolkitize(prob::DiffEqBase.SDEProblem)
                             return (prob.f.sys, prob.f.sys.states, prob.f.sys.ps)
     @parameters t
     var(x, i) = Sym{FnType{Tuple{symtype(t)}, Number}}(nameof(Variable(:x, i)))
-    vars = reshape([var(:x, i)(t) for i in eachindex(prob.u0)],size(prob.u0))
+    vars = reshape([var(:x, i)(value(t)) for i in eachindex(prob.u0)],size(prob.u0))
     params = prob.p isa DiffEqBase.NullParameters ? [] :
              reshape([Variable(:α,i) for i in eachindex(prob.p)],size(prob.p))
     @derivatives D'~t