arrays - 初始化非标准值的双精度数组的最快方法
问题描述
MATLAB 提供了用于预分配/初始化具有常见值(如0
或)的数组的函数1
。但是,如果我们希望数组具有任意double
值,则有多种方法可以做到这一点,并且不清楚哪一种更可取。
这个问题并不新鲜——之前在这篇博文和这个答案等地方已经讨论过了。然而,经验表明,软件(特别是 MATLAB 及其执行引擎)和硬件会随着时间而变化,因此在不同的系统上,最佳方法可能会有所不同。不幸的是,上述来源不提供基准测试代码,这可能是回答这个问题的终极(和永恒)方式。
考虑到我可能同时使用“常规”数组和各种大小的数组,我正在寻找一个可以运行的基准测试,它可以告诉我在我的系统上使用的最快方法。double
gpuArray double
解决方案
function allocationBenchmark(arrSz)
if nargin < 1
arrSz = 1000;
end
%% RAM
t = [];
disp('--------------- Allocations in RAM ---------------')
t(end+1) = timeit(@()v1(arrSz), 1);
t(end+1) = timeit(@()v2(arrSz), 1);
t(end+1) = timeit(@()v3(arrSz), 1);
t(end+1) = timeit(@()v4(arrSz), 1);
t(end+1) = timeit(@()v5(arrSz), 1);
t(end+1) = timeit(@()v6(arrSz), 1);
t(end+1) = timeit(@()v7(arrSz), 1);
t = 1E3 * t; % conversion to msec
disp(t); disp(" ");
[~,I] = min(t);
disp("Conclusion: method #" + I + " is the fastest on the CPU!"); disp(" ");
%% VRAM
if gpuDeviceCount == 0, return; end
t = [];
disp('--------------- Allocations in VRAM --------------')
t(end+1) = NaN; % impossible (?) to run v1 on the gpu
t(end+1) = gputimeit(@()v2gpu(arrSz), 1);
t(end+1) = gputimeit(@()v3gpu(arrSz), 1);
t(end+1) = gputimeit(@()v4gpu(arrSz), 1);
t(end+1) = gputimeit(@()v5gpu(arrSz), 1);
t(end+1) = gputimeit(@()v6gpu(arrSz), 1);
t(end+1) = gputimeit(@()v7gpu(arrSz), 1);
t = 1E3 * t; % conversion to msec
disp(t); disp(" ");
[~,I] = min(t);
disp("Conclusion: method #" + I + " is the fastest on the GPU!");
end
%% %%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%% RAM
function out = v1(M)
% Indexing on the undefined matrix with assignment:
out(1:M, 1:M) = pi;
end
function out = v2(M)
% Indexing on the target value using the `ones` function:
scalar = pi;
out = scalar(ones(M));
end
function out = v3(M)
% Using the `zeros` function with addition:
out = zeros(M, M) + pi;
end
function out = v4(M)
% Using the `repmat` function:
out = repmat(pi, [M, M]);
end
function out = v5(M)
% Using the ones function with multiplication:
out = ones(M) .* pi;
end
function out = v6(M)
% Default initialization with full assignment:
out = zeros(M);
out(:) = pi;
end
function out = v7(M)
% Using the `repelem` function:
out = repelem(pi,M,M);
end
%% %%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%% VRAM
function out = v2gpu(M)
scalar = gpuArray(pi);
out = scalar(gpuArray.ones(M));
end
function out = v3gpu(M)
out = gpuArray.zeros(M, M) + gpuArray(pi);
end
function out = v4gpu(M)
out = repmat(gpuArray(pi), [M, M]);
end
function out = v5gpu(M)
out = gpuArray.ones(M) .* gpuArray(pi);
end
function out = v6gpu(M)
% Default initialization with full assignment:
out = gpuArray.zeros(M);
out(:) = gpuArray(pi);
end
function out = v7gpu(M)
% Using the `repelem` function:
out = repelem(gpuArray(pi),M,M);
end
运行上述(例如,输入5000
)会导致以下结果:
--------------- Allocations in RAM ---------------
110.4832 328.1685 48.7895 47.9652 108.8930 93.0481 47.9037
Conclusion: method #7 is the fastest on the CPU!
--------------- Allocations in VRAM --------------
NaN 37.0322 17.9096 14.2873 17.7377 16.1386 16.6330
Conclusion: method #4 is the fastest on the GPU!
...它告诉我们在每种情况下使用的最佳(或等效)方法。
推荐阅读
- excel - 粘贴另一个工作表中的前导公共子字符串
- javascript - 如何在前端 Web 应用程序中隐藏测验的答案?
- ruby-on-rails - ActiveRecord 在双向线程中选择第一条消息
- excel - 使用 AGGREGATE 函数提取数组中的最大值会引发错误
- python - 显示 subprocess.Popen 的输出,就像在终端中运行一样
- flutter - 如何临时更改 Flutter 小部件中的文本颜色(并返回)?
- sql - 按另一列 SQL 中的唯一条目对列进行分组
- r - 在 R 中拆分向量并在列表中组织
- qb64 - 不移动时播放器图像消失/“load_map”加载速度不够快
- javascript - 如何从 iframe 按钮提交填充父页面内的文本输入?