cupy.fft.config.set_cufft_gpus#

cupy.fft.config.set_cufft_gpus(gpus)[source]#

Set the GPUs to be used in multi-GPU FFT.

Parameters:

gpus (int or list of int) – The number of GPUs or a list of GPUs to be used. For the former case, the first gpus GPUs will be used.

Warning

This API is currently experimental and may be changed in the future version.