Hello everybody!
We’re working on novel software for micromagnetic simulations.
And we wanna use Pytorch for all stuff.
My question is simple - is it possible to create a custom CUDA Kernel? I mean, if we wanna make some nontrivial tensor logic at only 1 kernel, how we can do it?
Thanks in advance,
Ruslan