DP3
Functions
IterativeDiagonal.h File Reference
#include <complex>
#include <cuda_runtime.h>
#include <cudawrappers/cu.hpp>

Go to the source code of this file.

Functions

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)
 

Function Documentation

◆ 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 
)