[tutorial] Adding New Compiler Backends to PyTorch JIT

jit
1

Hey all,

I wrote up a tutorial on how to integrate a new backend into PyTorch JIT. The process is relatively simple and only requires around 2-3 PyTorch specific API calls. It can be found here.

3 Likes
2

That looks like an awesome tutorial, Bram!
Thanks for sharing. :slight_smile:

3

It is very useful for me! Thanks for sharing!
I want to implement a third-party hardware backend compiler, is there any more examples or documents?
Thanks in advance!

4

Take a look at the torch/csrc/jit/backends/backend.h header. You can see an example of a backend registered using this API in the pytorch/glow repository.