cuFFT

Public

Private

cuFFT.cufftMakePlanMethod
cufftMakePlan(output_type::Type{<:cufftNumber}, input_type::Type{<:cufftNumber}, input_size::Dims, region)

low level interface to the CUDA library CuFFT for the function cufftXtMakePlanMany

Parameters:

  • output_type: type of the output array
  • input_type: type of the input array
  • input_size: size of the array to transform in units of the type
  • region: dimensions of the array to transform
source
cuFFT.get_batch_dimsMethod
get_batch_dims(region, sz)

returns the dimensions over which to run internal batching and dimensions used for external (for-loop) batching. It finds the largest product of consecutive dimensions and uses these as internal batch dimensions. All other dimensions are external batch dimensions.

internal_batch_dims, external_batch_dims = get_batch_dims(region, sz)

Parameters:

  • region: Tuple of dimensions to transform
  • sz: size of the array to transform. All dimensions not in region are considered as batch dimensions. This size Tuple is only used to determine the best set of consecutive dimensions to be used for internal batching.
source