Question

我在Fortran中编写了一个子程序来处理代码中计算密集的部分。我想使用mex函数将它链接到Matlab。这是给出相关错误的最简单版本。首先，这是mex功能。

#include <fintrf.h>
!======================================================================
#if 0
!     
!     example.F90
!     .F90 file needs to be preprocessed to generate .for equivalent
!     
#endif
!======================================================================
! Matlab command
! deltaHat = example(z)
!----------------------------------------------------------------------
!     Gateway routine
      SUBROUTINE mexFunction(nlhs, plhs, nrhs, prhs)
      IMPLICIT NONE

!     mexFunction arguments:
      MWPOINTER ::  plhs(*), prhs(*)
      INTEGER :: nlhs, nrhs

!     Function declarations:
      mwPointer :: mxGetPr
      mwPointer :: mxCreateDoubleMatrix
      mwSize :: mxGetM, mxGetN

!     Pointers to input/output mxArrays:
      mwPointer :: deltaHat
      mwPointer :: z

!     Array information:
      mwSize :: N

!------------------------------------------------------------------------
!     Get the size of the input array.
      N = mxGetN(prhs(1))

!     Create Fortran arrays from the input argument.
      z = mxGetPr(prhs(1))


!     Create matrix for the return argument.
      plhs(1) = mxCreateDoubleMatrix(1,24,0)
      deltaHat = mxGetPr(plhs(1))

!     Call the computational subroutine. 
      CALL example(%val(deltaHat), %val(z), N)

      END SUBROUTINE

!-----------------------------------------------------------------------
!     Computational routine for contraction mapping


      SUBROUTINE example(deltaHat, z, N)      
      IMPLICIT NONE

      INTEGER, PARAMETER :: dble = selected_real_kind(15)
      INTEGER, PARAMETER :: J = 24

      ! Type delcarations for inputs

      mwSize :: N
      REAL(KIND=dble), DIMENSION(1,J-1), INTENT(OUT) :: deltaHat
      REAL(KIND=dble), DIMENSION(N,1), INTENT(IN) :: z

      ! Declare local variables
      REAL(KIND=dble), DIMENSION(N,J) :: num

      num(:,1) = z(:,1)

      DO i = 1,J-1
        deltaHat(1,i) = num(i,1)
      END DO

      END SUBROUTINE

然后我使用“mex example.F90”在Matlab中编译mex函数。没有问题。现在这里是Matlab代码和输出。

clear all;

N = 1000;
J = 25;

% Generate covariates-----------------------------------------------------

% Reset random number generators                                
rng('default');

% Create indicators for parameter heterogeneity

z = randi(2,N,1);
z = z-ones(N,1);

deltaHat = example(z);

>> sum(deltaHat' - z(1:J-1))

ans =

    -15

>> deltaHat

deltaHat =

Columns 1 through 11

1.0000    0.0000         0         0    0.0000    0.0000    0.0000    0.0000    0.0000      0.0000    0.0000

Columns 12 through 22

0.0000    0.0000    0.0000    0.0000    0.0000    0.0000    0.0000    0.0000    0.0000    0.0000    0.0000

Columns 23 through 24

0.0000    0.0000

>> z(1:J-1)

ans =

 1
 1
 0
 1
 1
 0
 0
 1
 1
 1
 0
 1
 1
 0
 1
 0
 0
 1
 1
 1
 1
 0
 1
 1

术语sum（deltaHat' - z（1：J-1））应该为零，因为deltaHat和z（1：J-1）应该是相同的向量。我还在上面打印了这些向量，以防我发现了一些我缺少的变量类型定义（虽然我已经检查过它们都是双精度）。很明显，数据没有传递给Fortran并且正确地返回。但是，如果我在mex文件中进行以下调整

...
DO i = 1,J-1
   deltaHat(1,i) = z(i,1)
END DO
...

然后我得到了正确的答案。这是相应的Matlab输出。

...
>> sum(deltaHat' - z(1:J-1))

ans =

     0
...

因此，在这种情况下，数据将从Matlab传递到Fortran子例程并正确返回。问题似乎在于将z分配给mex文件中的局部变量num。跳过此步骤会给出正确的答案。

Answer 1

传递给N的

SUBROUTINE example为1.因此，当您指定z时，您正在触及数组deltaHat(1,2) = z(2,1)的越界

将行n = mxGetN(prhs(1))更改为n = mxGetM(prhs(1))

你应该学习如何debug your MEX functions，这样你就可以轻松解决这类琐碎的错误。

Fortran Mex文件 - 错误值传递回Matlab

1 个答案: