Skip to content

Commit 72486dc

Browse files
add lowering pass to converter test (#3820)
1 parent 622ce3a commit 72486dc

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed

tests/py/dynamo/conversion/harness.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -533,6 +533,9 @@ def run_test_compare_tensor_attributes_only(
533533
settings=compilation_settings,
534534
)
535535

536+
for pass_func in post_lowering_pass_list_for_converter_test:
537+
mod = pass_func(mod, compilation_settings)
538+
536539
interp = TRTInterpreter(
537540
mod,
538541
Input.from_tensors(inputs),
@@ -578,6 +581,8 @@ def run_test_with_dynamic_shape(
578581
settings=compilation_settings,
579582
torch_export_dynamic_shapes=torch_export_dynamic_shapes,
580583
)
584+
for pass_func in post_lowering_pass_list_for_converter_test:
585+
mod = pass_func(mod, compilation_settings)
581586

582587
if check_dtype:
583588
if use_dynamo_tracer:

0 commit comments

Comments
 (0)