Skip to content

Template for GPU accelerated python libraries

License

Notifications You must be signed in to change notification settings

prass-anyvision/pybind11-cuda

 
 

Repository files navigation

pybind11-cuda

Compiles out of the box with cmake

Numpy integration

C++ Templating for composable kernels with generic data types

Originally based on https://github.com/torstem/demo-cuda-pybind11

Prerequisites

Cuda installed in /usr/local/cuda

Python 3.6 or greater

Cmake 3.6 or greater

To build

source install.bash

Test it with python3 test_mul.py

About

Template for GPU accelerated python libraries

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Cuda 67.0%
  • CMake 15.6%
  • Python 14.2%
  • Shell 3.2%