Skip to content

Latest commit

 

History

History
11 lines (6 loc) · 566 Bytes

README.md

File metadata and controls

11 lines (6 loc) · 566 Bytes

HyRay

GPU ? CuPy : NumPy

GPU arrays (via CuPy) when possible, otherwise CPU arrays (via NumPy)

Code: https://github.com/static-frame/hyray

Packages: https://pypi.org/project/hyray

The hyray library provides a single interface to use both NumPy and CuPy, depending on the availability of CuPy, CUDA, or sufficient memory. With hyray a single implementation can run with CuPy when CuPy is available (on a local machine with a GPU), and still work with NumPy when on GPU or insufficient memory is available (such as on continuous integration servers).