def bench_mri_2d_16coils_fieldmap_multispectral(
navg_time=(1, 4),
plot_timings=False,
include_GPU=False,
include_CPU=True,
print_timings=True,
phasings=["real", "complex"],
save_dir=None,
nspectra=2,
):
recon_cases = []
if include_GPU:
recon_cases += ["GPU,Sp", "GPU,Tab"]
precisions = ["single"]
else:
precisions = ["single", "double"]
if include_CPU:
recon_cases += ["CPU,Tab", "CPU,Sp"]
# spectral_offsets = [0, 434.316]
spectral_offsets = np.asarray(
[0, 434.316, 332.124, 485.412, -76.644, 49.8186, 247.8156]
)
spectral_offsets = spectral_offsets[:nspectra]
t = _test_mri_multi(
ndim=2,
N0=64,
grid_os_factor=1.5,
J0=4,
Ld=4096,
n_coils=16,
fieldmap_segments=6,
precisions=precisions,
phasings=phasings,
recon_cases=recon_cases,
compare_to_exact=False,
navg_time=4,
spectral_offsets=spectral_offsets,
)
t_create, t_forward, t_adj, t_norm = _split_timings(t, print_timings)
if plot_timings:
title_string = "2D (16 coils, 6 fieldmap seg, %d spectral)" % nspectra
axes = _plot_all_timings(
t_create, t_forward, t_adj, t_norm, title_substring=title_string
)
else:
axes = None
if save_dir is not None:
fname_base = "bench_2d_016coil_006fmapseg_%dspectra" % nspectra
_save_bench(
save_dir, fname_base, [t_create, t_forward, t_adj, t_norm], axes
)
def bench_mri_3d_16coils_fieldmap(
navg_time=(1, 4),
plot_timings=False,
include_GPU=False,
include_CPU=True,
print_timings=True,
phasings=["real", "complex"],
save_dir=None,
):
recon_cases = []
if include_GPU:
recon_cases += ["GPU,Sp", "GPU,Tab"]
precisions = ["single"]
else:
precisions = ["single", "double"]
if include_CPU:
recon_cases += ["CPU,Tab", "CPU,Sp"]
t = _test_mri_multi(
ndim=3,
N0=64,
grid_os_factor=1.5,
J0=4,
Ld=4096,
n_coils=16,
fieldmap_segments=6,
precisions=precisions,
phasings=phasings,
recon_cases=recon_cases,
compare_to_exact=False,
navg_time=navg_time,
)
t_create, t_forward, t_adj, t_norm = _split_timings(t, print_timings)
if plot_timings:
axes = _plot_all_timings(
t_create,
t_forward,
t_adj,
t_norm,
title_substring="3D (16 coils, 6 fieldmap seg)",
)
else:
axes = None
if save_dir is not None:
fname_base = "bench_3d_016coil_006fmapseg"
_save_bench(
save_dir, fname_base, [t_create, t_forward, t_adj, t_norm], axes
)
