Custom Vulkan / GLSL Extension tutorial


I have seen the “Custom C++ and CUDA Extensions” tutorial here:

However, it doesn’t cover Vulkan / GLSL yet. Is expanding the tutorial for Vulkan / GLSL planned?

In the meantime, would anyone have pointers on how to best do extend PyTorch with a GLSL shader similar to those already in the main repo (pytorch/aten/src/ATen/native/vulkan/glsl at master · pytorch/pytorch · GitHub)?

Many thanks,