Skip to content

Commit 4694c69

Browse files
author
Wei
authored
Update config.yml (#1163)
* Update config.yml * Update test_graph_opts.py can not use prefix `test` for non-test func.
1 parent 2717771 commit 4694c69

File tree

2 files changed

+3
-2
lines changed

2 files changed

+3
-2
lines changed

.circleci/config.yml

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -49,6 +49,7 @@ jobs:
4949
pip3 install nvidia-tensorrt==8.2.4.2
5050
pip3 install --pre torch==1.13.0.dev20220621 torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/nightly/cu113
5151
pip3 install pytest parameterized expecttest
52+
pip3 install tabulate
5253
# install torch_tensorrt
5354
mv WORKSPACE.ci WORKSPACE
5455
cd py

py/torch_tensorrt/fx/test/passes/test_graph_opts.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -18,7 +18,7 @@ def debug_print_graph_module(mod_graph: torch.fx.GraphModule) -> None:
1818

1919

2020
@torch.fx.wrap
21-
def test_op(keys, value):
21+
def _test_op(keys, value):
2222
return value
2323

2424

@@ -170,7 +170,7 @@ def forward(self, a, b, c):
170170
def test_common_subexpression_elimination_string_arg(self):
171171
class TestModule(torch.nn.Module):
172172
def forward(self, a):
173-
x = test_op(["foo", "bar"], a)
173+
x = _test_op(["foo", "bar"], a)
174174
return x
175175

176176
self._test_opt_with_module(

0 commit comments

Comments
 (0)