search

inducer / pycuda

CUDA integration for Python, plus shiny features
http://mathema.tician.de/software/pycuda
Other
1.82k stars 286 forks source link

Vulkan interoperability #435

Open lgleim opened 5 months ago

lgleim commented 5 months ago

Is your feature request related to a problem? Please describe. OpenGL rendering is increasingly replaced by the Vulkan API. Interoperability between Cuda and Vulkan is cumbersome and requires native programming as in these Cuda samples:

  1. https://github.com/NVIDIA/cuda-samples/tree/master/Samples/5_Domain_Specific/simpleVulkan
  2. https://github.com/NVIDIA/cuda-samples/tree/master/Samples/5_Domain_Specific/vulkanImageCUDA
  3. https://github.com/NVIDIA/cuda-samples/tree/master/Samples/5_Domain_Specific/simpleVulkanMMAP

Describe the solution you'd like A python interface to interface pytorch with vulkan rendering through CUDA, similar to the OpenGL renderer of pytorch3d.

inducer commented 5 months ago

I would be happy to consider a PR along those lines.