← back

pytorch-op-deps: Hand-rolled LLVM parser for PyTorch operator analysis

A static analyzer that parses LLVM IR to build dependency graphs of PyTorch operators. Hand-wrote the LLVM IR parser to trace which operators depend on which C++ kernels.

The motivation: when you're trying to shrink PyTorch for mobile/embedded deployment, you need to know exactly which operators pull in which code. This tool maps those dependencies so you can strip out what you don't need.

github.com/ljk53/pytorch-op-deps