t-vi opened a new pull request #10794: URL: https://github.com/apache/tvm/pull/10794
This bumps PyTorch to 1.11 and fixes 3 test failures. The bump is required to enable the libtorch_ops fallback due to DLPack version incompatibilities. QAT training has its own `fuse_modules` version (`fuse_modules_qat`) in PyTorch, so I changed the test. Two amendments to the front end: - `searchsorted` gets more (optional) parameters to its signature, - There is a `sub` variant with `alpha` (`a - alpha * b`). PyTorch rewrites `rsub` with `alpha` to this, but we ignored it. Now we handle `sub` with alpha. Thank you, @masahi for getting me started with the bump and pointing out the test failures. Any errors are my own. -- 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]
