knswd的博客

GPU篇前提上次讲过了如何让写的eigen代码跑在线程池上，这次讲讲跑在gpu上的版本。当然我们讨论的类都为/unsupported/Eigen/CXX11/ 中定义的类。也是tensorflow许多算子gpu版本实现的依赖。看源码老样子看一下gpustream和gpudevice的定义首先了解一下StreamInterface这个类，从名字就可以看出来这个类作为基类定义了各种流的接口，主要是为了规范各种stream的行为（通过纯虚函数的形式逼迫子类实现这些stream基本行为，比如内存管理）。

前提虽说Eigen是一个优秀的并行计算库，但是只跑在单线程环境下在牛鼻和你我写出来的代码也没啥区别。本文需要读者了解GPU并行计算的基础。先看会源码大伙先来看源码：有句话说的好，源码之前了无秘密/usr/local/include/unsupported/Eigen/CXX11这个文件夹下的头文件将满足你的所有需求(嘿嘿，起飞咯)先看一下Tensor头文件：#ifdef EIGEN_USE_THREADS#include "ThreadPool"#endif#ifdef EIGEN_

下载https://gitlab.com/libeigen/eigen/-/releases/3.3.9tar -xvf 解压解压后进入文件夹