File tree Expand file tree Collapse file tree 1 file changed +8
-0
lines changed
tests/py/dynamo/conversion Expand file tree Collapse file tree 1 file changed +8
-0
lines changed Original file line number Diff line number Diff line change 26
26
post_lowering ,
27
27
pre_export_lowering ,
28
28
)
29
+ from torch_tensorrt .dynamo .lowering .passes import remove_num_users_is_0_nodes
29
30
from torch_tensorrt .dynamo .runtime import PythonTorchTensorRTModule
30
31
from torch_tensorrt .dynamo .utils import ATOL , RTOL , get_model_device , get_torch_inputs
31
32
32
33
_LOGGER : logging .Logger = logging .getLogger (__name__ )
33
34
35
+ # this is the post lowering pass list for the converter test
36
+ post_lowering_pass_list_for_converter_test = [
37
+ remove_num_users_is_0_nodes ,
38
+ ]
39
+
34
40
35
41
# this method is only used in our converter test to infer the module output dtypes via dummy inference
36
42
# which is due to fx.symbolic_trace does not have the meta['val'] info in the node
@@ -435,6 +441,8 @@ def run_test(
435
441
settings = compilation_settings ,
436
442
)
437
443
444
+ for pass_func in post_lowering_pass_list_for_converter_test :
445
+ mod = pass_func (mod , compilation_settings )
438
446
num_inputs = len (inputs )
439
447
trt_inputs = inputs
440
448
dtype_to_change = []
You can’t perform that action at this time.
0 commit comments