stepan
/
dynare
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

block_trust_region MEX: add support for sparse Jacobian 

If the function to be solved returns a sparse Jacobian, simply convert it to a
dense representation.
unit-tests
Sébastien Villemot 2022-11-30 14:47:48 +01:00
parent 3e391347d4
commit 10fdc42516
No known key found for this signature in database
GPG Key ID: 2CECE9350ECEBE4A
1 changed files with 12 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
mex/sources/block_trust_region/matlab_fcn_closure.F08
View File

@ -134,11 +134,21 @@ contains
! Handle Jacobian
if (present(fjac)) then
if (.not. mxIsDouble(call_lhs(2)) .or. mxIsSparse(call_lhs(2)) &
if (.not. mxIsDouble(call_lhs(2)) &
.or. mxGetM(call_lhs(2)) /= n_all .or. mxGetN(call_lhs(2)) /= n_all) &
call mexErrMsgTxt("Second output argument of the function must be a dense double float&
call mexErrMsgTxt("Second output argument of the function must be a double float&
& square matrix whose dimension matches the length of the first input argument")
! If Jacobian is sparse, convert it to a dense matrix
if (mxIsSparse(call_lhs(2))) then
block
type(c_ptr) :: dense_jacobian
if (mexCallMATLAB(1_c_int, dense_jacobian, 1_c_int, call_lhs(2), "full") /= 0) &
call mexErrMsgTxt("Error converting Jacobian from sparse to dense representation")
call_lhs(2) = dense_jacobian
end block
end if
if (.not. mxIsComplex(call_lhs(2))) then ! Real case
block
real(real64), dimension(:,:), pointer, contiguous :: fjac_all