在Matlab中通过mex使用cuda的方法

最新推荐文章于 2022-10-24 15:14:25 发布

苹果挨炮

最新推荐文章于 2022-10-24 15:14:25 发布

阅读量1.6k

点赞数 2

分类专栏： matlab cuda 文章标签： matlab c++

本文链接：https://blog.csdn.net/m0_56015193/article/details/126171031

版权

matlab 同时被 2 个专栏收录

1 篇文章 0 订阅

订阅专栏

cuda

1 篇文章 0 订阅

订阅专栏

仅供参考，具体问题需要具体分析！

1.cuda安装及配置，参考下面这篇

cpp文件调用cuda函数_苹果挨炮的博客-CSDN博客

2.Matlab配置mex编译环境

在matlab命令行中输入以下命令

mex  -setup

然后会提示用户是安装c++版本的还是fortran版本的

这里我们选择c++版本的，输入以下命令

mex -setup C++

然后matlab会在你的电脑里检查可用编译器，提示用户选择不同的编译器安装（假如你的电脑里安装的是2022版的visula c++，但你的matlab版本低于2022版，可能会检测不出来）

这里我们不选择2022版本，选择2017版本的（原因后面讲），输入提示的命令

mex -setup:'D:\Program Files\MATLAB\R2022a\bin\win64\mexopts\msvcpp2022.xml' C++

3.编写.cpp文件并用mex编译

在介绍可被mexcuda编译的.cu文件之前，先说明一下可被mex编译的.cpp文件的编写方式，与普通的.cpp文件不同，这里的.cpp文件中没有main()函数，而是mexFuncion()函数，它大概长这个样子

# include "mex.h"

void mexFunction(int nlhs, mxArray *plhs[],int nrhs,const mxArray *prhs[])
{   
    //nlhs (number of left-hand side) 是输出参数的个数
    //nrhs (number of right-hand side) 是输入参数的个数
    //prhs[]是指针数组，指向输入的参数，这个参数是在matlab中调用编译好的mexw64时输入的参数
    //如phrs[0]指向第一个参数，phrs[1]指向第二个参数
    //plhs[]也是指针数组，指向输出的变量
}

以下是一段求两点之间斜段斜率的函数

# include "mex.h"

void mexFunction(int nlhs, mxArray *plhs[],int nrhs,const mxArray *prhs[])
{   
    //mxGetScalar用于获取第一个参数的值
    double x1 = mxGetScalar(prhs[0]);
    double y1 = mxGetScalar(prhs[1]);
    double x2 = mxGetScalar(prhs[2]);
    double y2 = mxGetScalar(prhs[3]);
    //假如你第一个参数是个矩阵，需要用mxGetPr()来获取
    //double *data = mxGetPr(prhs[0]);
    double k = (y2 - y1) / (x2 - x1);
    double b = (x2 * y1 - x1 * y2) / (x2 - x1);
    //mxCreateDoubleMatrix(m, n, _)用来创建一个大小为m×n的矩阵用来储存,_指定有无虚部
    plhs[0] = mxCreateDoubleMatrix(1, 2, mxREAL);
    double* res = mxGetPr(plhs[0]);
    //plhs[0]相当于老板，负责揽项目(开辟内存)，再把揽好的项目交给工程队去做(将内存地址
    //赋值给res，再用res去赋值)
    res[0] = k;
    res[1] = b;
}

写完后在matlab中用mex xx.cpp编译，然后就可以当做函数使用了