贤者之路，Cuda block内部矩阵求逆，mxm矩阵复杂度为O(m)

在做线性变换上经常要用到NXN的矩阵求逆。在CUDA用的是高斯消元比较适合并行计算。

下面是3X3Cuda实现矩阵求逆的Device函数，也就是说可以直接写到你的kernel函数上去。当然也可以是任何NXN矩阵

另外时间上，测试过6X6的矩阵用时在0.016ms，所以可以说基本不耗时。以后会把速度的梯度测试和精度测试结果都放上去。

*   1   2   1   | 1 0   0    1   0   0 | -2.5   1.5   0.5
*   2   3   1   | 0 1   0       =>      0   1   0 |   1.5 -0.5 -0.5
*   1   1   2   | 0 0   1    0   0   1 |   0.5 -0.5   0.5

* 输入矩阵 E E 逆矩阵

//3X3InvMatrix MX --> MXI

//input MX[3][3]
//.............
//.............

__shared__ double MXI[3][3];

int isx = threadIdx.x;
int isy = threadIdx.y;
double tmpIn;
double tmpInv;
//initialize E
if(isx == isy)
MXI[isy][isx] = 1;
else
MXI[isy][isx] = 0;

for (int i = 0; i < 3; i++)
{
  if (i == isy && isx < 3 && isy < 3)
  {
     //消除对角线上的元素（主元）为1
   tmpIn = MX[i][i];
     MX[i][isx] /= tmpIn;
   MXI[i][isx] /= tmpIn;
   }
__syncthreads();
if (i != isy && isx < 3 && isy < 3)
{
   //将主元所在列的元素化为0 所在行的元素同时变化
    tmpInv = MX[isy][i];
    MX[isy][isx] -= tmpInv * MX[i][isx];
    MXI[isy][isx] -= tmpInv * MXI[i][isx];
   }
   __syncthreads();
}

当然还可以将速度提一倍，就是开更多的线程和资源。然后MX 和 MXI一起算

贤者之路，Cuda block内部矩阵求逆，mxm矩阵复杂度为O(m)相关推荐

c语言block内部的实现原理,iOS中block变量捕获原理详析
Block概述 Block它是C语言级别和运行时方面的一个特征.Block封装了一段代码逻辑,也用{}括起,和标准C语言中的函数/函数指针很相似,此外就是blokc能够对定义环境中的变量可以引用到.这 ...
hdu 2243 考研路茫茫——单词情结(AC自动+矩阵)
考研路茫茫--单词情结 Time Limit: 2000/1000 MS (Java/Others) Memory Limit: 32768/32768 K (Java/Others) Tota ...
矩阵求逆操作的复杂度分析（逆矩阵的复杂度分析）
矩阵求逆操作的复杂度分析逆矩阵的复杂度分析 1 背景之前写过一篇关于矩阵复杂度分析的文章,没有想到阅读人数那么多.对于IT相关人士来说,从代码层次再结合基本数学知识,就能够很好地理解矩阵的复杂度如 ...
矩阵求逆，矩阵转置，矩阵相乘
矩阵求逆,矩阵转置,矩阵相乘矩阵求逆用的高斯消元 #include <bits/stdc++.h>using namespace std;const double eps = 1e-8; ...
贤者之路，cuda版本convertto实现（与OPENCV 3.4 CPU版本数值一致）
[引言]: 将一个float32精度的矩阵砍到uchar精度,每个库都会根据自己算法目标类型做一些加速的优化从而导致结果不一样,比如在OpenCV3.4 cpu版本的convertto中, 1.5f的 ...
贤者之路， Caffe转TensorRT
[引]将Caffe转TensorRT的时候,有很多自己设计的接口TensorRT库本身不支持.我们需要自己创建Plugin,本文介绍TensorRT的创建,如何自定义Plugin,和快速书写cuda函 ...
贤者之路，Tensorrt的int8 calibration创建
[INT8 Calibration] 无论哪块GPU计算板卡都能够支持Int8的加速,但需要事先生成 calibration文件,下面是.h .cpp分别继承IInt8EntropyCalibrato ...
hdu_2243_考研路茫茫——单词情结(AC自动机+矩阵)
题目链接:hdu_2243_考研路茫茫--单词情结题意: 让你求包含这些模式串并且长度不小于L的单词种类题解: 这题是poj2788的升级版,没做过的强烈建议先做那题. 我们用poj2778的方法 ...
Python——矩阵求逆、矩阵的转置
一.前提介绍设A是数域上的一个n阶方阵,若在相同数域上存在另一个n阶矩B,使得: AB=BA=E. 则我们称B是A的逆矩阵,而A则被称为可逆矩阵.其中,E为单位矩阵. 数学上求解方法可以通过:初等变 ...

贤者之路，Cuda block内部矩阵求逆，mxm矩阵复杂度为O(m)

贤者之路，Cuda block内部矩阵求逆，mxm矩阵复杂度为O(m)相关推荐

最新文章

热门文章

贤者之路，Cuda block内部矩阵求逆，mxm矩阵 复杂度为O(m)

贤者之路，Cuda block内部矩阵求逆，mxm矩阵 复杂度为O(m)相关推荐

最新文章

热门文章

贤者之路，Cuda block内部矩阵求逆，mxm矩阵复杂度为O(m)

贤者之路，Cuda block内部矩阵求逆，mxm矩阵复杂度为O(m)相关推荐