Skip to content
/ pybind11-cuda Public template

Template for GPU accelerated python libraries

License

Notifications You must be signed in to change notification settings

PWhiddy/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

Topics

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published