Need to fix cupy types - Githubissues

gallantlab / himalaya

Multiple-target linear models - CPU/GPU

https://gallantlab.github.io/himalaya

BSD 3-Clause "New" or "Revised" License

78 stars 13 forks source link

Need to fix cupy types #55

Closed mvdoc closed 5 months ago

mvdoc commented 5 months ago

With thanks to @gxlilyBerkeley for the bug report.

We need to update some functions in the cupy backend (e.g., changing cupy.bool to cupy.bool_ in https://github.com/gallantlab/himalaya/blob/main/himalaya/backend/cupy.py#L114). Unfortunately the tests are not run on github because of the lack of GPU support. So I'm running the test locally to determine what to change.