diff --git a/.circleci/config.yml b/.circleci/config.yml index 0223e697dc..903e33d4b7 100644 --- a/.circleci/config.yml +++ b/.circleci/config.yml @@ -49,6 +49,7 @@ jobs: pip3 install nvidia-tensorrt==8.2.4.2 pip3 install --pre torch==1.13.0.dev20220621 torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/nightly/cu113 pip3 install pytest parameterized expecttest + pip3 install tabulate # install torch_tensorrt mv WORKSPACE.ci WORKSPACE cd py diff --git a/py/torch_tensorrt/fx/test/passes/test_graph_opts.py b/py/torch_tensorrt/fx/test/passes/test_graph_opts.py index f240c95514..1f852b0497 100644 --- a/py/torch_tensorrt/fx/test/passes/test_graph_opts.py +++ b/py/torch_tensorrt/fx/test/passes/test_graph_opts.py @@ -18,7 +18,7 @@ def debug_print_graph_module(mod_graph: torch.fx.GraphModule) -> None: @torch.fx.wrap -def test_op(keys, value): +def _test_op(keys, value): return value @@ -170,7 +170,7 @@ def forward(self, a, b, c): def test_common_subexpression_elimination_string_arg(self): class TestModule(torch.nn.Module): def forward(self, a): - x = test_op(["foo", "bar"], a) + x = _test_op(["foo", "bar"], a) return x self._test_opt_with_module(