permute in GPU

7 visualizzazioni (ultimi 30 giorni)
Rodrigo
Rodrigo il 25 Ago 2011
I am trying to do a 3D fft on some data using the built in GPU toolbox of 2011a and am having some trouble. Since the fftn function doesn't exist I find myself needing to do fft2 followed by fft. However, since "permute" doesn't seem to exist for GPU arrays either, I have to reshape the [l m n] matrix into a [l*m n] matrix, transposing it, applying fft, transposing back and reshaping back. Is there a more elegant/less asinine way of doing this?

Risposta accettata

Jan
Jan il 25 Ago 2011
RESHAPE is very efficient, because it does not touch the data, but only the vector of dimensions. Therefore the computational costs of your procedure are optimal already, if the TRANSPOSE is programmed efficiently. A PERMUTE cannot be faster.
I'd expect, that süpecifying the dimension to operate on is helpful here: "fft(X,[],dim)". But I don't know, if this works with the GPU toolbox.

Più risposte (2)

Walter Roberson
Walter Roberson il 25 Ago 2011
I seem to recall John saying that Jacket supports fftn(); if you are doing heavy GPU work, you may wish to investigate Accelereyes Jacket .

Rodrigo
Rodrigo il 29 Ago 2011
The reshape command works very fast. The .' operation needed to shift the dimensions not so much. The routine spends about 20% of it's time doing that, but it's still much faster than the CPU version.
If my department has a site license for Jacket I'll give it a shot. Otherwise I don't want to spend the extra money for such a simple operation.

Tag

Community Treasure Hunt

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

Start Hunting!

Translated by