Skip to content

Commit 67b454f

Browse files
Wei Weifacebook-github-bot
Wei Wei
authored andcommitted
[fx2trt] support ops for hf_T5 (#62)
Summary: X-link: pytorch/fx2trt#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
1 parent b75fbe3 commit 67b454f

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

torch/testing/_internal/common_fx2trt.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -208,7 +208,7 @@ def run_test(
208208

209209
if test_implicit_batch_dim:
210210
interp = TRTInterpreter(mod, InputTensorSpec.from_tensors(inputs))
211-
super().run_test(mod, inputs, expected_ops, unexpected_ops, interp, rtol, atol)
211+
super().run_test(mod, inputs, expected_ops, unexpected_ops, interp, rtol, atol, precision)
212212

213213
if test_explicit_batch_dim:
214214
interp = TRTInterpreter(

0 commit comments

Comments
 (0)