denizyuret
AutoGrad and CUDA kernels
Are there any plans to support custom CUDA kernels? This would be amazing.
I think this should be possible right now, as long as both the forward and the backward operations are implemented. See the Knet package (and the custom kernels in its libknet8.so library) for some examples.
On Wed, Aug 3, 2022 at 10:45 AM Renato Bellotti @.***> wrote:
Are there any plans to support custom CUDA kernels? This would be amazing.
— Reply to this email directly, view it on GitHub https://github.com/denizyuret/AutoGrad.jl/issues/128, or unsubscribe https://github.com/notifications/unsubscribe-auth/AAN43JXCKVMADD7MWG5CE4TVXIPTLANCNFSM55NZUQBQ . You are receiving this because you are subscribed to this thread.Message ID: @.***>
