juda opened a new pull request, #11911: URL: https://github.com/apache/tvm/pull/11911
The pull request contains two functions: 1. `optimize_torch` as a function similar to `torch.jit.trace`, which is used to optimize the `torch.nn.module` by TVM metaSchedule. 2. `as_torch` as a decorator, which is used to wrap the TVMscript code to `torch.nn.module`. The files consist of: + Two python codes for two functions and a C++ backend + Two unit tests for both functions + Two simple tutorials showing the usage. @yelite @junrushao1994 @masahi -- 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]
