Add support for batched eigh

desihub / gpu_specter

Scratch work for porting spectroperfectionism extractions to GPUs

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

2 stars 3 forks source link

Closed dmargala closed 3 years ago

dmargala commented 4 years ago

This PR adds support for batched eigh on the gpu. If the batched eigh function is not available in cupy, it falls back to the non-batched eigh.