标题: MATLAB矩阵相乘源程序 [打印本页]
作者: lynzdh    时间: 2019-1-12 10:43
标题: MATLAB矩阵相乘源程序
matlab 矩阵相乘


全部资料51hei下载地址：
mtimesx_20110223.zip (255.8 KB, 下载次数: 7)
  1. /* OpenMP ------------------------------------------------------------- */

  3. #ifdef _OPENMP
  4. #include <omp.h>
  5. #endif

  7. /* Includes ----------------------------------------------------------- */

  9. #include <string.h>
  10. #include <stddef.h>
  11. #include <ctype.h>
  12. #include <math.h>
  13. #include <time.h>
  14. #include "mex.h"

  16. /* Macros ------------------------------------------------------------- */

  18. #ifndef MWSIZE_MAX
  19. #define  mwIndex        int
  20. #define  mwSignedIndex  int
  21. #define  mwSize         int
  22. #define  mwSize_t       size_t
  23. #else
  24. #define  mwSize_t       mwSize
  25. #endif

  27. #define  METHOD_BLAS       0
  28. #define  METHOD_LOOPS      1
  29. #define  METHOD_LOOPS_OMP  2

  31. #define  MTIMESX_NOT_SET    0
  32. #define  MTIMESX_BLAS       1
  33. #define  MTIMESX_LOOPS      2
  34. #define  MTIMESX_LOOPS_OMP  3
  35. #define  MTIMESX_MATLAB     4
  36. #define  MTIMESX_SPEED      5
  37. #define  MTIMESX_SPEED_OMP  6

  39. #define  STRINGIZE(name)  #name
  40. #define  TOKENSTRING(name)  STRINGIZE(name)

  42. #define  DIRECTIVE_MAX  1000

  44. #ifndef  COMPILER
  45. #define  COMPILER  (unknown)
  46. #endif

  48. /* Prototypes --------------------------------------------------------- */

  50. mxArray *DoubleTimesDouble(mxArray *, char, mxArray *, char);
  51. mxArray *FloatTimesFloat(mxArray *, char, mxArray *, char);
  52. mxArray *mxCreateSharedDataCopy(const mxArray *pr);
  53. char mxArrayToTrans(const mxArray *mx);
  54. void *myRealloc(void *vp, mwSize_t n);
  55. void mtimesx_logo(void);
  56. mxArray *modestring(int);

  58. /* Global Variables --------------------------------------------------- */

  60. int mtimesx_mode  = MTIMESX_MATLAB;
  61. int max_threads   = 0;
  62. int threads_used  = 0;
  63. int debug         = 0;
  64. int debug_message = 0;

  66. /* Functions ect for OpenMP implementations ------- */

  68. #ifdef _OPENMP

  70. #define  OPENMP_ENABLED   1.0

  72. /* Functions etc for non OpenMP implementations ----------------- */

  74. /* The omp_get_num_procs() function is courtesy of Dirk-Jan Kroon */
  75. /* and is based on his FEX submission maxNumCompThreads --------- */

  77. #else

  79. #define  OPENMP_ENABLED   0.0
  80. #define  omp_get_wtick()  0.0
  81. #define  omp_get_wtime()  ((double)clock()/((double)CLOCKS_PER_SEC))

  83. #if defined(_WIN32) || defined(_WIN64)

  85. #include <windows.h>
  86. int omp_get_num_procs( ) {
  87.     SYSTEM_INFO sysinfo;
  88.     GetSystemInfo(&sysinfo);
  89.     return sysinfo.dwNumberOfProcessors;
  90. }

  92. #elif MACOS

  94. #include <sys/param.h>
  95. #include <sys/sysctl.h>
  96. int omp_get_num_procs( ) {
  97.     int nm[2];
  98.     size_t len = 4;
  99.     uint32_t count;

  101.     nm[0] = CTL_HW; nm[1] = HW_AVAILCPU;
  102.     sysctl(nm, 2, &count, &len, NULL, 0);

  104.     if(count < 1) {
  105.         nm[1] = HW_NCPU;
  106.         sysctl(nm, 2, &count, &len, NULL, 0);
  107.         if(count < 1) { count = 1; }
  108.     }
  109.     return count;
  110. }
  111. #else

  113. #include <unistd.h>
  114. int omp_get_num_procs( ) {
  115.     return sysconf(_SC_NPROCESSORS_ONLN);
  116. }

  118. #endif

  120. #endif

  122. /*------------------------------------------------------------------------ */
  123. /*------------------------------------------------------------------------ */
  124. /*------------------------------------------------------------------------ */
  125. /*------------------------------------------------------------------------ */

  127. void mexFunction(int nlhs, mxArray *plhs[],int nrhs, const mxArray *prhs[])
  128. {
  129.     mxArray *A, *B, *C, *Araw, *Braw;
  130.     mxArray *rhs[4];
  131.     char transa, transb;
  132.     char *directive, *cp;
  133.     char transstring[2] = "_";
  134.     int unsupported = 0;
  135.     int i, j, k, got_directive, nrhs_old = nrhs;
  136.         int mtimesx_mode_new, max_threads_new,
  137.                 already_set_mode, already_set_debug, already_set_threads;
  138.         mxArray *ans;
  139.         double threads;

  141. /*-------------------------------------------------------------------------
  142. * Check for proper number of inputs and outputs
  143. *------------------------------------------------------------------------- */

  145.     if( nlhs > 1 ) {
  146.         mexErrMsgTxt("Must have at most 1 output.");
  147.     }

  149. /*-------------------------------------------------------------------------
  150. * If no inputs, just return the current mode
  151. *------------------------------------------------------------------------- */

  153.     if( nrhs == 0 ) {
  154.                 plhs[0] = modestring(mtimesx_mode);
  155.         return;
  156.     }

  158. /*-------------------------------------------------------------------------
  159. * Find out if any input is a directive
  160. *------------------------------------------------------------------------- */

  162.         i = 0;
  163.         mtimesx_mode_new    = MTIMESX_NOT_SET;
  164.         max_threads_new     = 0;
  165.         already_set_mode    = 0;
  166.         already_set_debug   = 0;
  167.         already_set_threads = 0;
  168.         while( i < nrhs ) {
  169.         if( mxIsChar(prhs[i]) ) {
  170.                 if( mxGetNumberOfElements(prhs[i]) > DIRECTIVE_MAX ) {
  171. ……………………

  173. …………限于本文篇幅 余下代码请从51黑下载附件…………
复制代码






欢迎光临 (http://www.51hei.com/bbs/) Powered by Discuz! X3.1