Fast gpuArray slicing for cart2sph
1 次查看(过去 30 天)
显示 更早的评论
I have a 3 x N gpuArray, where N is very large, dimension 1 represents x, y, and z, and these points are the result of a viewpoint transformation applied via multiplication of a 3 x N matrix with a 3x3 rotation matrix and sum with a translation vector. At this point I need to convert to spherical coordinates, however to do this I have to slice the array into its x, y, and z components before passing these separate arrays as inputs to cart2sph (or sub functions like atan2 if I want to write my own version).
The problem is that the slicing of the array to feed into cart2sph takes much longer than the viewpoint transformation and the coordinate transform combined. The only way I've found to accelerate this is to replace the slicing with a dot-multiply-and-sum operation, which for some reason is faster than simply slicing. Here's some example code:
tst = randn(3, 6000000, 'single', 'gpuArray');
tic
for n = 1:100
tst1 = tst(1, :);
tst2 = tst(2, :);
tst3 = tst(3, :);
[th, phi, r] = cart2sph(tst1, tst2, tst3);
end
wait(gpuDevice);
toc
tic
for n = 1:100
tst1b = sum(tst.*[1;0;0]);
tst2b = sum(tst.*[0;1;0]);
tst3b = sum(tst.*[0;0;1]);
[th, phi, r] = cart2sph(tst1b, tst2b, tst3b);
end
wait(gpuDevice);
toc
On my computer the first loop takes ~1.5 seconds and the second ~0.5. A dot-multiply & sum is 3x faster than simply slicing the array, and the cart2sph takes only a trivial amount of time. So my questions are:
1) Is there a faster way to get from a 3xN xyz array to a 3xN phi-theta-r array, preferably that does not require a set of (very slow) slicing operations?
2) Why is a multiply and sum operation faster than a simple slicing operation?
Thank you
0 个评论
采纳的回答
Edric Ellis
2020-7-15
Like all arrays in MATLAB, gpuArray data is stored in "column-major" order. One consequence of this is that it is much more efficient to extract individual columns from a matrix than individual rows - this is true on the CPU, and doubly true on the GPU. Extracting a column is equivalent to a simple memory block copy. Extracting a row requires a "strided" copy operation. You can take advantage of this by performing a single up-front transpose on your array, and then use efficient column indexing operations:
tst_t = tst.';
for n = 1:100
tst1 = tst_t(:, 1).';
tst2 = tst_t(:, 2).';
tst3 = tst_t(:, 3).';
[th, phi, r] = cart2sph(tst1, tst2, tst3);
end
On my machine with a now rather old Tesla K20c, this approach takes 0.015 seconds compared with 5.3 seconds for your first approach and 2.0 seconds for your second approach. (Note that transposing vectors can be extremely effficient because the memory layout is identical)
更多回答(0 个)
另请参阅
类别
在 Help Center 和 File Exchange 中查找有关 Logical 的更多信息
产品
Community Treasure Hunt
Find the treasures in MATLAB Central and discover how the community can help you!
Start Hunting!