Why doesn't MATLAB support 'cpuArray' for some array creation functions?

2 vues (au cours des 30 derniers jours)
For example, we can create a GPU array by code:
rand(3, 'single', 'gpuArray')
Unfortunately, this code will fail if it is run on a PC without a CUDA graphics card. If we can pass 'cpuArray' to rand, we can easily write a MATLAB program for both CPU and GPU computings. E.g.,
useGpu = 0
if(useGpu)
arrayType = 'gpuArray';
else
arrayType = 'cpuArray';
end
% The program ...
A = rand(3, 'single', arrayType);
B = rand(3, 'single', arrayType);
C = A*B;
(Unfortunately, this example doesn't work for current MATLAB...)
Otherwise, we have to write this code:
useGpu = 0
if(useGpu)
A = rand(3, 'single', 'gpuArray');
B = rand(3, 'single', 'gpuArray');
else
A = rand(3, 'single');
B = rand(3, 'single');
end
C = A*B;
Could MathWorks add 'cpuArray' to most array creation functions in the future?

Réponse acceptée

Edric Ellis
Edric Ellis le 17 Jan 2018
Another workaround is to use a cell array.
if useGpu
arg = {'gpuArray'};
else
arg = {};
end
A = rand(3, 'single', arg{:});
The {:} syntax converts arg into a comma-separated list.
  1 commentaire
Meng-Yuan Huang
Meng-Yuan Huang le 17 Jan 2018
Modifié(e) : Meng-Yuan Huang le 17 Jan 2018
Thanks. Your workaround is better, because we can specify data type for each array.

Connectez-vous pour commenter.

Plus de réponses (1)

Meng-Yuan Huang
Meng-Yuan Huang le 17 Jan 2018
I find a workaround:
useGpu = 1;
if(useGpu)
arg1 = 'single';
arg2 = 'gpuArray';
else
arg1 = 1;
arg2 = 'single';
end
% The program ...
A = rand(3, 3, arg1, arg2);
B = rand(3, 3, arg1, arg2);
C = A*B

Catégories

En savoir plus sur GPU Computing dans Help Center et File Exchange

Tags

Community Treasure Hunt

Find the treasures in MATLAB Central and discover how the community can help you!

Start Hunting!

Translated by