Is cuDNN currently used for the convolution kernel on the NVIDIA platform? How does it work for AMD GPUs or others?
Modular is not using cuDNN at all. That’s the whole point and USP of the stack.
The kernels are written in Mojo – you can find them in the Modular repository on GitHub.
1 Like