您的位置:首页 > 编程语言

CUDA的cublas 和 Intel的MKL 矩阵运算对比

2015-08-20 17:14 375 查看
CUBLAS和MKL都是快速矩阵运算的工具 。   一个适用Intel的cpu,一个适用于nvidia的GPU。

最近在做RNN(循环神经网络)的加速。 其中一点就是把神经网络的矩阵运算放到CPU上算。所以就做了一点相关的测试。

以前我们实验室的RNN用的运算库是MKL   改到CUDA上加速之后,发现不仅没有加速,速度还变慢了!

一直以为是代码写错了。 查了好久才发现是因为 : mkl适合频繁小矩阵相乘 cuda适合不频繁大矩阵相乘

这是我做的测试:

cublas & mkl 速度小测试: (float) :

 matrix_size:10; calc_times:10^7matrix_size:100; calc_times:1000000matrix_size:1000; calc_times:1000matrix_size:10000; calc_times:1matrix_size:30000; calc_times:1
mkl5.231s26.73s7.089s7.088s2m20.647s
cublas2m0.327s33.608s4.255s3.821s2.887s (测了无数次,3W的比1W的时间还少
可以发现,基本所当矩阵到达几百的时候,cublas和mkl的速度能持平     之后就是矩阵越大cublas的计算速度越快。,否则则是mkl表现更优秀。

因为RNN在做语言模型的时候隐藏层只有200嘛一般。 因此基本上都是一些小矩阵运算。 所以反而比CPU上MKL慢。

~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~

大家在进行GPU加速的时候一定要考虑矩阵大小这个因素哦。
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息