Skip to content

annemariegagn/cublas-magma-pybind11

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

8 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

pycuda

GPU linear algebra functions callable from Python

Example how to use high-performance GPU libraries such as Cublas or Magma with Python using Pybind11

Pybind11: https://pybind11.readthedocs.io/en/stable/

Magma: http://icl.cs.utk.edu/magma/

Create a simple Python module that can be imported in any project. CPU libraries Eigen are also supported, as well as the possibility to add Cuda kernels in .cu files. Requirements and precisions can be found in the CMakeLists.txt file provided that can be adapted to other Linux configurations.

Further support for cublas functions, as well as automatic Python-side documentation will be added.

Anne-Marie G.

About

GPU linear algebra functions callable from Python

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published