Quelle bibliothèque utilisez-vous pour les calculs matriciels sur CUDA?

Question

Quelle bibliothèque utilisez-vous pour les calculs matriciels sur CUDA? Ou y en a-t-il? Il semble que tout le monde écrit cela tout seul.

Pour les processeurs habituels, j'utilise Eigen. Qu'en est-il des GPU?

Answer 1

Pour les opérations de matrice dense, vous pouvez prendre en compte CUBLAS (fourni avec CUDA Toolkit), Magma et CULAtools.

Pour les opérations à matrice clairsemée, veuillez prendre en compte CUSPARSE (fourni avec CUDA Toolkit) et CUSP.

Answer 2

Quels sont les problèmes que vous rencontrez?
Pour les calculs de matrice clairsemée, CUSP est une très bonne bibliothèque. Pour les problèmes denses Magma peut être un meilleur ajustement.