Skip to content

Offload phasing to GPU using CuPy

Iris Chang requested to merge iris_dev_cupy_ into development

This is a replacement for !17 (closed) for offloading phasing to GPU using CuPy. Changes include: (1) add pip install --no-cache-dir cupy in setup/build_from_scratch.sh; (2) add use_cupy in spinifel/settings.py; (3) replace using_cuda with use_cuda; (4) add use_cupy under [runtime] in toml file; (5) replace NumPy with CuPy in sequential/phasing.py. Note that one needs to export CUPY_CACHE_DIR=$PWD/setup/cupy before the jsrun command.

Edited by Iris Chang

Merge request reports