Update

Author: odow

src/Nonlinear/ReverseAD/utils.jl

@@ -53,11 +53,11 @@ function Base.setindex!(x::_UnsafeVectorView{T}, value::T, i::Integer) where {T}
     return value
 end
 
-function Base.setindex!(
-    x::_UnsafeVectorView{T},
-    value::T,
-    i::CartesianIndex{1},
-) where {T}
+function Base.setindex!(x::_UnsafeVectorView{T}, value, i::Integer) where {T}
+    return setindex!(x, convert(T, value), i)
+end
+
+function Base.setindex!(x::_UnsafeVectorView, value, i::CartesianIndex{1})
     return setindex!(x, value, i[1])
 end
 

test/Nonlinear/ReverseAD.jl

@@ -1331,6 +1331,38 @@ function test_eval_user_defined_operator_ForwardDiff_gradient!()
     return
 end
 
+function test_eval_user_defined_operator_type_mismatch()
+    model = MOI.Nonlinear.Model()
+    x = MOI.VariableIndex.(1:4)
+    p = MOI.Nonlinear.add_parameter(model, 2.0)
+    ex = MOI.Nonlinear.add_expression(model, :($p * $(x[1])))
+    ψ(x) = sin(x)
+    t(x, y) = x + 3y
+    function ∇t(ret, x, y)
+        ret[1] = 1      # These are intentionall the wrong type
+        ret[2] = 3 // 1 # These are intentionall the wrong type
+        return
+    end
+    MOI.Nonlinear.register_operator(model, :ψ, 1, ψ, x -> -cos(x))
+    MOI.Nonlinear.register_operator(model, :t, 2, t, ∇t)
+    MOI.Nonlinear.add_constraint(
+        model,
+        :($ex^3 + sin($(x[2])) / ψ($(x[2])) + t($(x[3]), $(x[4]))),
+        MOI.LessThan(0.0),
+    )
+    d = MOI.Nonlinear.Evaluator(model, MOI.Nonlinear.SparseReverseMode(), x)
+    MOI.initialize(d, [:Jac])
+    X = [1.1, 1.2, 1.3, 1.4]
+    g = [NaN]
+    MOI.eval_constraint(d, g, X)
+    @test only(g) ≈ 17.148
+    @test MOI.jacobian_structure(d) == [(1, 1), (1, 2), (1, 3), (1, 4)]
+    J = [NaN, NaN, NaN, NaN]
+    MOI.eval_constraint_jacobian(d, J, X)
+    @test J ≈ [2.0^3 * 3.0 * 1.1^2, 0.0, 1.0, 3.0]
+    return
+end
+
 end  # module
 
 TestReverseAD.runtests()