File tree Expand file tree Collapse file tree 1 file changed +6
-5
lines changed
tests/py/dynamo/conversion Expand file tree Collapse file tree 1 file changed +6
-5
lines changed Original file line number Diff line number Diff line change 11import torch
22import torch .nn as nn
3+ import torch_tensorrt
34from parameterized import parameterized
45from torch .testing ._internal .common_utils import TestCase , run_tests
5- import torch_tensorrt
66
77rand_ops = [
88 (
@@ -75,13 +75,14 @@ def forward(self):
7575 return self .rand_op (self .size )
7676
7777 grid_model = TestModule (op , shape_or_input )
78- #cannot use self.run_test() since it expects input in form of tensor
79-
80- #self.run_test(grid_model, None)
78+ # cannot use self.run_test() since it expects input in form of tensor
79+
80+ # self.run_test(grid_model, None)
8181 fx_graph = torch .fx .symbolic_trace (grid_model )
8282 torch ._dynamo .reset ()
8383
84- optimized_model = torch_tensorrt .compile (fx_graph ,
84+ optimized_model = torch_tensorrt .compile (
85+ fx_graph ,
8586 "torch_compile" ,
8687 None ,
8788 min_block_size = 1 ,
You can’t perform that action at this time.
0 commit comments