News for hackers. Stuff that matters.

A comment on VkFFT: Vulkan/CUDA/Hip/OpenCL/Level Zero/Metal Fast Fourier Transform Library. Parent comment

Re:VkFFT: Vulkan/CUDA/Hip/OpenCL/Level Zero/Metal Fast Fouri... by Y_Y on Wednesday August 02, 2023 @04:59AM (#36968631)
Implementing a custom layer isn't hard[0], that said if it were me I'd rather add it in the runtime, e.g. via a TensorRT plugin.

[0] e.g. https://jamesmccaffrey.wordpress.com/2021/09/02/example-of-a...

[ Parent | on HN ]
Re:VkFFT: Vulkan/CUDA/Hip/OpenCL/Level Zero/Metal Fast Fouri... by mathisfun123 on Wednesday August 02, 2023 @09:57AM (#36970982)
That tut only shows you how to remix existing operators. That obviously won't work here (not what's required). You need to do this

https://pytorch.org/tutorials/advanced/dispatcher.html

More complex but still not that hard.

[ Parent | on HN ]