PyTorch Developer Podcast
PyTorch Developer Podcast
Edward Yang, Team PyTorch
How new operators are authored
15 minutes Posted May 18, 2021 at 1:00 pm.
0:00
15:33
Download MP3
Show notes

What's the general process by which a new operator is added to PyTorch? Why is this actually something of a rare occurrence? How do you integrate an operator with the rest of PyTorch's system so it can be run end-to-end? What should I expect if I'm writing a CPU and CUDA kernel? What tools are available to me to make the job easier? How can I debug my kernels? How do I test them?

Further reading.