Skip to content

Offload phasing to GPU using Cupy

Iris Chang requested to merge iris_dev_cupy into development

This is 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) add use_cupy under [runtime] in toml file; (3) 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