Matlab: Does calling the same mex function repeatedly from a loop incur too much overhead?

后端 未结 3 862
伪装坚强ぢ
伪装坚强ぢ 2021-01-18 00:30

I have some Matlab code which needs to be speeded up. Through profiling, I\'ve identified a particular function as the culprit in slowing down the execution. This function i

3条回答
  •  生来不讨喜
    2021-01-18 01:31

    As usual, it all depends on the amount of work you do in the MEX file.. The overhead of calling MEX function is constant and does not depend on e.g., the problem size. It means that arguments are not copied to new, temporary arrays. Hence, if it is enough work, the MATLAB overhead of calling the MEX file will not show. Anyway, in my experience the MEX call overhead is significant only for the first time the mex function is called - the dynamic library has to be loaded, symbols resolved etc. Subsequent MEX calls have very little overhead and are very efficient.

    Almost everything in MATLAB is connected with some overhead due to the nature of this high-level language. Unless you have a code, which you are sure is fully compiled with JIT (but then you do not need a mex file :)) So you have a choice of one overhead over the other..

    So sum up - I would not be too scared of MEX calling overhead.

    Edit As often heard here and elsewhere, the only reasonable thing to do in any particular case is of course BENCHMARK and check it for your self. You can easily estimate the MEX call overhead by writing a trivial MEX function:

    #include "mex.h"
    void mexFunction(int nlhs, mxArray *plhs[ ], int nrhs, const mxArray *prhs[ ]) 
    {      
    }
    

    On my computer you get

    tic; for i=1:1000000; mexFun; end; toc
    Elapsed time is 2.104849 seconds.
    

    That is 2e-6s overhead per MEX call. Add your code, time it and see, if the overhead is at acceptable level, or not.

    As Andrew Janke noted below (thanks!), the MEX function overhead apparently depends on the number of arguments you pass to the MEX function. It is a small dependence, but it is there:

    a = ones(1000,1);
    tic; for i=1:1000000; mexFun(a); end; toc
    Elapsed time is 2.41 seconds.
    

    It is not related to size of a:

    a = ones(1000000,1);
    tic; for i=1:1000000; mexFun(a); end; toc
    Elapsed time is 2.41805 seconds.
    

    But it is related to the number of arguments

    a = ones(1000000,1);
    b = ones(1000000,1);
    tic; for i=1:1000000; mexFun(a, b); end; toc
    Elapsed time is 2.690237 seconds.
    

    So you might want to take that into account in your tests.

提交回复
热议问题