AutoGrad and CUDA kernels
renatobellotti opened this issue · 1 comments
renatobellotti commented
Are there any plans to support custom CUDA kernels? This would be amazing.
denizyuret commented
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
<#128>, or unsubscribe
<https://github.com/notifications/unsubscribe-auth/AAN43JXCKVMADD7MWG5CE4TVXIPTLANCNFSM55NZUQBQ>
.
You are receiving this because you are subscribed to this thread.Message
ID: ***@***.***>