pytorch
f323a8ab - [fx2trt] support ops for hf_T5 (#62)

Commit
2 years ago
[fx2trt] support ops for hf_T5 (#62) Summary: X-link: https://github.com/pytorch/fx2trt/pull/62 1. support type_as() 2. improve power with method 3. improve to_dtype(torch.Tensor) 4. support float(),half(),int() 5. support isinf(). Not possible to support isnan() 6. support torch.any() 7. support rsqrt() (Note: this ignores all push blocking failures!) Test Plan: Test passed Reviewed By: 842974287 Differential Revision: D35516820 fbshipit-source-id: 7f4b7c1b833034121f7643565bd4d1b4133319bf (cherry picked from commit 67b454f91dcad0678b57d78dae14cce21d32b84a)
Author
Wei Wei
Committer
Parents
Loading