#include <complex>
#include <cuda_runtime.h>
#include <cudawrappers/cu.hpp>
Go to the source code of this file.
|
| void | LaunchSolveDirectionKernel (cudaStream_t stream, size_t n_visibilities, size_t n_direction_solutions, size_t n_solutions, size_t direction, cu::DeviceMemory &antenna_pairs, cu::DeviceMemory &solution_map, cu::DeviceMemory &solutions, cu::DeviceMemory &model, cu::DeviceMemory &residual_in, cu::DeviceMemory &residual_temp, cu::DeviceMemory &numerator, cu::DeviceMemory &denominator) |
| |
| void | LaunchSolveNextSolutionKernel (cudaStream_t stream, size_t n_antennas, size_t n_visibilities, size_t n_direction_solutions, size_t n_solutions, size_t direction, cu::DeviceMemory &antenna_pairs, cu::DeviceMemory &solution_map, cu::DeviceMemory &next_solutions, cu::DeviceMemory &numerator, cu::DeviceMemory &denominator) |
| |
| void | LaunchStepKernel (cudaStream_t stream, size_t n_visibilities, cu::DeviceMemory &solutions, cu::DeviceMemory &next_solutions, bool phase_only, double step_size) |
| |
| void | LaunchSubtractKernel (cudaStream_t stream, size_t n_directions, size_t n_visibilities, size_t n_solutions, cu::DeviceMemory &antenna_pairs, cu::DeviceMemory &solution_map, cu::DeviceMemory &solutions, cu::DeviceMemory &model, cu::DeviceMemory &residual) |
| |
◆ LaunchSolveDirectionKernel()
| void LaunchSolveDirectionKernel |
( |
cudaStream_t |
stream, |
|
|
size_t |
n_visibilities, |
|
|
size_t |
n_direction_solutions, |
|
|
size_t |
n_solutions, |
|
|
size_t |
direction, |
|
|
cu::DeviceMemory & |
antenna_pairs, |
|
|
cu::DeviceMemory & |
solution_map, |
|
|
cu::DeviceMemory & |
solutions, |
|
|
cu::DeviceMemory & |
model, |
|
|
cu::DeviceMemory & |
residual_in, |
|
|
cu::DeviceMemory & |
residual_temp, |
|
|
cu::DeviceMemory & |
numerator, |
|
|
cu::DeviceMemory & |
denominator |
|
) |
| |
◆ LaunchSolveNextSolutionKernel()
| void LaunchSolveNextSolutionKernel |
( |
cudaStream_t |
stream, |
|
|
size_t |
n_antennas, |
|
|
size_t |
n_visibilities, |
|
|
size_t |
n_direction_solutions, |
|
|
size_t |
n_solutions, |
|
|
size_t |
direction, |
|
|
cu::DeviceMemory & |
antenna_pairs, |
|
|
cu::DeviceMemory & |
solution_map, |
|
|
cu::DeviceMemory & |
next_solutions, |
|
|
cu::DeviceMemory & |
numerator, |
|
|
cu::DeviceMemory & |
denominator |
|
) |
| |
◆ LaunchStepKernel()
| void LaunchStepKernel |
( |
cudaStream_t |
stream, |
|
|
size_t |
n_visibilities, |
|
|
cu::DeviceMemory & |
solutions, |
|
|
cu::DeviceMemory & |
next_solutions, |
|
|
bool |
phase_only, |
|
|
double |
step_size |
|
) |
| |
◆ LaunchSubtractKernel()
| void LaunchSubtractKernel |
( |
cudaStream_t |
stream, |
|
|
size_t |
n_directions, |
|
|
size_t |
n_visibilities, |
|
|
size_t |
n_solutions, |
|
|
cu::DeviceMemory & |
antenna_pairs, |
|
|
cu::DeviceMemory & |
solution_map, |
|
|
cu::DeviceMemory & |
solutions, |
|
|
cu::DeviceMemory & |
model, |
|
|
cu::DeviceMemory & |
residual |
|
) |
| |