0

我有一个 mex 函数，它采用双精度输入向量，将精度向下转换为临时单精度向量，进行一些处理，然后将处理结果再次向上转换为双精度。

下面的简化代码示例编译并说明了这个过程。

#include "mex.h"

void mexFunction(int nlhs, mxArray *plhs[], int nrhs, const mxArray *prhs[])
{           
  int xM = mxGetM(prhs[0]); 
  int xN = mxGetN(prhs[0]);         
  int totalNumElements = xM*xN;       

  plhs[0] = mxCreateDoubleMatrix(xM,xN,mxREAL);

  double* out  = mxGetPr(plhs[0]); 
  double* in   = mxGetPr(prhs[0]);    

  float out32[totalNumElements]; 
  float  in32[totalNumElements]; 

  // ---------> DOWNCAST <---------
  for (int mm = 0; mm < totalNumElements; ++mm)
      in32[mm] = (float)in[mm];
  // ---------- DOWNCAST ----------

  // Do some single precision signal processing (just copy for this example)
  for (int mm = 0; mm < totalNumElements; ++mm)
      out32[mm] = in32[mm];

  // ---------> UPCAST <---------
  for (int mm = 0; mm < totalNumElements; ++mm)
      out[mm] = (double)out32[mm];
  // ---------- UPCAST ---------- 
}

在我的机器上，像这样调用已编译的 mex 函数可以正常工作。.

>> x = randn(1e6,1); y=demo(x);

...但是这样调用它会导致 Matlab 意外关闭

>> x = randn(1e7,1); y=demo(x);

看到崩溃是由输入向量的大小增加引起的，我猜测错误是由于内存分配失败造成的。如果发生此类错误，我如何优雅地退出到 Matlab 并给出错误消息？谢谢。

4

1 回答 1

2

方法一

好的，一种方法是用 C-style 替换堆栈分配的变量。. .

 float*  in32  =  (float*)mxCalloc(totalNumElements, sizeof(float));
 float*  out32 =  (float*)mxCalloc(totalNumElements, sizeof(float));

当出现内存错误时，Matlab 巧妙地给出如下错误。. .

使用演示时出错 内存不足。

键入 HELP MEMORY 作为您的选项。

不需要try-catch块或异常。

方法二

另一种方法是使用 C++ 风格std::exception。堆栈分配的变量可以替换为以下...

  float*  in32;
  float*  out32;
  try
  {
      in32  = new float[totalNumElements];
      out32 = new float[totalNumElements];
  }
  catch (std::exception& e)
  {
      std::string msg = std::string("Aw snap: ") + e.what();
      mexErrMsgTxt(msg.c_str());
  }

当出现内存错误时，Matlab 巧妙地给出如下错误。. .

使用演示时出错

噢快照：std::bad_alloc

笔记

方法 1 更紧凑，更好地与 mex API 集成。最好在 mex 网关函数中使用方法 1，在库中使用方法 2，这可能不仅仅用于制作 Matlab 扩展。这两种方法都要求您使用mxFree( mxCallocsolution) 或delete( newsolution) 来收取在堆上为in32和分配的内存费用out32。我仍然不知道如何检测或处理堆栈溢出

于 2012-05-16T13:36:25.257 回答