为什么MATLAB在矩阵乘法中如此之快？

作者: 狗头军师
发布时间: 2025-03-09 02:07:42 (1天前)

2 条回复

1#
回复此人
v-star*위위 | 2020-08-14 14-37

这是在装有Tesla C2070的计算机上使用MATLAB R2011a + 并行计算工具箱的结果： ``` >> A = rand(1024); gA = gpuArray(A); % warm up by executing the operations a couple of times, and then: >> tic, C = A * A; toc Elapsed time is 0.075396 seconds. >> tic, gC = gA * gA; toc Elapsed time is 0.008621 seconds. ``` MATLAB使用高度优化的库进行矩阵乘法，这就是为什么简单的MATLAB矩阵乘法如此之快的原因。该gpuArray版本使用MAGMA。在配备Tesla K20c的计算机上使用R2014a进行更新，以及新功能timeit和gputimeit功能： ``` >> A = rand(1024); gA = gpuArray(A); >> timeit(@()A*A) ans = 0.0324 >> gputimeit(@()gA*gA) ans = 0.0022 ``` 在具有16个物理核心和Tesla V100的WIN64计算机上使用R2018b更新： ``` >> timeit(@()A*A) ans = 0.0229 >> gputimeit(@()gA*gA) ans = 4.8019e-04 ``` （注意：在某些时候（我忘记了确切的时间）gpuArray从MAGMA切换到cuBLAS- gpuArray尽管MAGMA仍用于某些操作）

编辑

登录后才能参与评论