[GUIDE] Getting C++ custom ops to work with torch.compile

I believe you’re looking for Black box a Python function for use with torch.compile - Google Docs