from galario import single_cuda as g_single
else:
from galario import double as g_double
from galario import single as g_single
# PARAMETERS FOR MULTIPLE TEST EXECUTIONS
par1 = {'dRA': 0., 'dDec': 0.4, 'PA': 2., 'nxy': 1024}
par2 = {'dRA': -3.5, 'dDec': 7.2, 'PA': -23., 'nxy': 2048}
par3 = {'dRA': 2.3, 'dDec': 3.2, 'PA': 88., 'nxy': 4096}
par4 = {'dRA': 0., 'dDec': 0., 'PA': 145., 'nxy': 1024}
# use last gpu if available. Check `watch -n 0.1 nvidia-smi` to see which gpu is
# used during test execution.
ngpus = g_double.ngpus()
g_double.use_gpu(0) #max(0, ngpus-1))
g_double.threads()
########################################################
# #
# TESTS #
# #
########################################################
@pytest.mark.parametrize("Rmin, dR, nrad, nxy, dxy, inc, profile_mode, real_type",
[(1e-6, 0.001, 2000, 1024, 0.2, 20., 'Gauss', 'float64'),
(1e-6, 0.001, 2000, 2048, 0.2, 44.23, 'Cos-Gauss', 'float64'),
(1e-6, 0.001, 2000, 2048, 0.5, 20., 'Gauss', 'float64'),
(1e-6, 0.001, 2000, 1024, 0.3, 20., 'Gauss', 'float64')],
default=False,
help="If True uses Python implementation of sample/chi2 Profile/Image.")
p.add_argument("--no-verbose",
action="store_true",
default=False,
help="Print timing to stdout")
options = p.parse_args()
# print(options)
if options.USE_GPU:
if galario.HAVE_CUDA:
from galario import double_cuda as acc_lib_cuda
acc_lib_cuda.use_gpu(options.gpu_id)
else:
print("Option --gpu not valid. galario.HAVE_CUDA is {}.".format(
galario.HAVE_CUDA))
options.USE_GPU = False
from galario import double as acc_lib_cpu
def setup_chi2Image(nxy, nsamples):
# these number can be freely changed for this speed test
dRA = -3.1
dDec = 2.5
PA = 80.