Open masalim2 opened 3 years ago
cupy.linalg.eigh may be a drop-in replacement for numpy.linalg.eigh and yield big boost:
cupy.linalg.eigh
numpy.linalg.eigh
https://docs.cupy.dev/en/stable/reference/generated/cupy.linalg.eigh.html
@braceal
cupy.linalg.eighmay be a drop-in replacement fornumpy.linalg.eighand yield big boost:https://docs.cupy.dev/en/stable/reference/generated/cupy.linalg.eigh.html
@braceal