How to load custom TorchScript ops in PyTorch Mobile?

dalalaa · January 1, 2020, 9:05am

Is there any tutorials or examples about loading custom TorchScript ops in PyTorch Mobile

dalalaa · January 7, 2020, 1:49am

@David_Reiss @IvanKobzarev Could I load custom TorchScript ops in PyTorch Mobile?

jspisak · January 7, 2020, 6:24am

what are the custom ops you are trying to script? If the model is scriptable you shouldnt need to do anything else (unless I am missing something).

tom · January 7, 2020, 7:40am

If you create a library (.so) that features your custom op registration as a static variable and link that into your app, it’ll work. That’s how I did it back in 2018 and it should still work.

Best regards

Thomas

David_Reiss · January 10, 2020, 12:49am

Thomas is correct. Here’s a demo of using the C++ API directly, which should allow custom op registration: https://github.com/ljk53/pytorch-android-cpp-demo

I know the process is a bit cumbersome. We’re looking at simplifying this over time.