pycuda
pycuda copied to clipboard
Vulkan interoperability
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:
- https://github.com/NVIDIA/cuda-samples/tree/master/Samples/5_Domain_Specific/simpleVulkan
- https://github.com/NVIDIA/cuda-samples/tree/master/Samples/5_Domain_Specific/vulkanImageCUDA
- 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.
I would be happy to consider a PR along those lines.