Add `__matmul__` to the magic methods for FX tracing (#64512)
authorHorace He <chilli@fb.com>
Wed, 8 Sep 2021 16:59:04 +0000 (09:59 -0700)
committerFacebook GitHub Bot <facebook-github-bot@users.noreply.github.com>
Wed, 8 Sep 2021 17:03:48 +0000 (10:03 -0700)
commit35413a16f7433b345ba1dc9c460cbe9c2d975762
tree30cd5faac2d2ab8682518294a76d6bfa9ab4e101
parent195cb4efa8b204ee1c9cb908308d516b812790e2
Add `__matmul__` to the magic methods for FX tracing (#64512)

Summary:
Fixes https://github.com/pytorch/pytorch/issues/64483

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

Reviewed By: mrshenli

Differential Revision: D30797265

Pulled By: Chillee

fbshipit-source-id: 7630e048a960e0b27c4309d04d85301abe325189
test/test_fx.py
torch/fx/graph.py