Meteorix opened a new pull request #8777: URL: https://github.com/apache/tvm/pull/8777
To increase the TVM accessibility for PyTorch users, we add PyTorchTVM module to support the following workflow: 1. convert a torchscript module to tvm graph 2. build and tune tvm graph 3. export well-tuned tvm graph as a pytorch op 4. torch jit trace the tvm pytorch op with other pytorch modules, then save/load/serve as normal pytorch model The example usage is here: [apps/pt_class/tests/test_pt_script.py](https://github.com/Meteorix/tvm/blob/meteorix_main_2/apps/pt_class/tests/test_pt_script.py). We hope to further discuss the user api with the community. Please help review @Laurawly @junrushao1994 @tqchen, thanks! Credit: the original author is @kongroo . -- This is an automated message from the Apache Git Service. To respond to the message, please log on to GitHub and use the URL above to go to the specific comment. To unsubscribe, e-mail: [email protected] For queries about this service, please contact Infrastructure at: [email protected]
