TylerADavis opened a new pull request #7128: URL: https://github.com/apache/tvm/pull/7128
Thanks for contributing to TVM! Please refer to guideline https://tvm.apache.org/docs/contribute/ for useful information and tips. After the pull request is submitted, please request code reviews from [Reviewers](https://github.com/apache/incubator-tvm/blob/master/CONTRIBUTORS.md#reviewers) by @ them in the pull request thread. This PR adds support for the `is_floating_point` and `div_` PyTorch operations, and closes #6239 . This PR does not include a unit test for `div_` as there is currently no unit test for `div`, and they share an implementation. Additionally, while I wrote a unit test for `is_floating_point()` (see https://github.com/apache/tvm/compare/main...TylerADavis:tyler_add_ops_incl_tests?expand=1), it is not included in this PR as models consisting solely of `is_floating_point` are compiled to constant graphs with no inputs, and the current testing infrastructure assumes that all graphs have inputs. The unit test for `is_floating_point()` tests the operator with `torch.jit.script` and `torch.jit.trace` with inputs consisting of `int8, float64, float32`, and `float16`. Each of these test cases passes. ---------------------------------------------------------------- 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. For queries about this service, please contact Infrastructure at: [email protected]
