Fix error of test_save_optimized_model_pass unittest (#56522)

29efbdb6 · Yuanle Liu · GitHub · b9330fe1 · 29efbdb6
隐藏空白更改
内联并排

Showing with 3 addition and 3 deletion

test/ir/inference/test_save_optimized_model_pass.py test/ir/inference/test_save_optimized_model_pass.py +3 -3

未找到文件。
--- a/test/ir/inference/test_save_optimized_model_pass.py
+++ b/test/ir/inference/test_save_optimized_model_pass.py
@@ -122,9 +122,9 @@ class TestSaveOptimizedModelPassWithTRT(
                use_calib_mode=False,
            )
            config.set_trt_dynamic_shape_info(
-                {"x": [1, 3, 224, 224], "flatten_1.tmp_0": [1, 9216]},
-                {"x": [1, 3, 224, 224], "flatten_1.tmp_0": [1, 9216]},
-                {"x": [1, 3, 224, 224], "flatten_1.tmp_0": [1, 9216]},
+                {"x": [1, 3, 224, 224], "flatten_0.tmp_0": [1, 9216]},
+                {"x": [1, 3, 224, 224], "flatten_0.tmp_0": [1, 9216]},
+                {"x": [1, 3, 224, 224], "flatten_0.tmp_0": [1, 9216]},
            )
            config.exp_disable_tensorrt_ops(["flatten_contiguous_range"])
            config.enable_memory_optim()