gpu_dedisperse

your.utils.gpu.gpu_dedisperse(cand, device=0)

GPU dedispersion (by rolling the array)

Args:

cand: Candidate instance

device (int): GPU ID

Returns:

candidate object

gpu_dmt

your.utils.gpu.gpu_dmt(cand, device=0)

GPU DM-Time bow-tie (by rolling the array)

Args:

cand: Candidate instance

device (int): GPU ID

Returns:

candidate object

gpu_dedisp_and_dmt_crop

your.utils.gpu.gpu_dedisp_and_dmt_crop(cand, device=0)

GPU based dedispersion, DM time bow-time plot and crop it to 256x256 shaped arrays (by rolling the array)

Args:

cand: Candidate instance

device (int): GPU ID

Returns:

candidate object

get_gpu_memory_map

your.utils.gpu.get_gpu_memory_map(gpu_id)

Get the current gpu free memory

Args:

gpu_id (int): GPU id

Returns:

int: amount of free GPU RAM