Home

Awesome

nvrtc.torch

Torch7 bindings for CUDA NVRTC (runtime compilation) library http://docs.nvidia.com/cuda/nvrtc/index.html

Requires CUDA 7 installation.

Example of usage:

local nvrtc = require 'nvrtc'
local ptx = nvrtc.compileReturnPTX(kernel)

where kernel is a lua string with CUDA kernel code. Returned PTX can be loaded and ran with CUDA Driver API. More examples coming.