dft_gpu_raw_kernel

ska_sdp_func_python.imaging.dft.dft_gpu_raw_kernel(direction_cosines, uvw_lambda, vfluxes)[source]

CPU computational kernel for DFT, using CUDA raw code via cupy.

Parameters:
  • direction_cosines – Direction cosines [ncomp, 3]

  • vfluxes – Fluxes [ncomp, nchan, npol]

  • uvw_lambda – UVW in lambda [ntimes, nbaselines, nchan, 3]

Returns:

Vis array, dims: [ntimes, nbaselines, nchan, npol]