projects / platform / upstream / pytorch.git / commit
? search:
summary | shortlog | log | commit | commitdiff | tree
(parent: b9fc656) | patch
Add `BinaryUfuncOpInfo` and broadcasting tests (#61964)
authorPhilip Meier <github.pmeier@posteo.de>
Fri, 20 Aug 2021 18:43:07 +0000 (11:43 -0700)
committerFacebook GitHub Bot <facebook-github-bot@users.noreply.github.com>
Fri, 20 Aug 2021 18:44:15 +0000 (11:44 -0700)
commit70a3210ecaa0162b4673f53faa17675a9d3ca8de
tree377833e9b3334a13eea76298b0a726363491504ctree | snapshot
parentb9fc656cf26d60127bd695e4e5a7d27622f2563dcommit | diff
Add `BinaryUfuncOpInfo` and broadcasting tests (#61964)

Summary:
As proof of concept, this PR uses the new `BinaryUfuncOpInfo` in broadcasting tests for `add`, `sub`, `mul`, `div`, `floor_div`, and `true_div`.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/61964

Reviewed By: ngimel

Differential Revision: D30407734

Pulled By: mruberry

fbshipit-source-id: ada28994f43b0635f279f45a02ecba18bc8ee033
test/test_binary_ufuncs.py diff | blob | history
test/test_jit_fuser_te.py diff | blob | history
torch/testing/_internal/common_methods_invocations.py diff | blob | history
Domain: Machine Learning / ML Framework;