ut

Author: xadupre

_unittests/ut_export/test_dynamic_shapes.py

@@ -710,7 +710,7 @@ def test_couple_input_ds_replace_by_string(self):
             {
                 "X": {0: "batch"},
                 "A": {0: "batch", 1: "Dim1"},
-                "B": ({0: "batch", 1: "Dim1"}, {0: "batch", 1: "Dim1"}),
+                "B": ({0: "batch", 1: "Dim2"}, {0: "batch", 1: "Dim3"}),
             },
             res,
         )

onnx_diagnostic/export/dynamic_shapes.py

@@ -148,7 +148,6 @@ def _replace_dim_tensor_by_string(cls, inputs, ds, unique: Set[str]):
         new_ds = ds.copy()
         for i, v in ds.items():
             if isinstance(v, str):
-                assert v not in unique, f"Dimension {v!r} is already defined in {unique}"
                 unique.add(v)
                 new_ds[i] = v
             elif v in (torch.export.Dim.DYNAMIC, torch.export.Dim.AUTO):

onnx_diagnostic/torch_models/test_helper.py

@@ -1183,7 +1183,7 @@ def run_ort_fusion(
             optimization_options=opts,
         )
     except Exception as e:
-        duration = {time.perf_counter() - begin}
+        duration = time.perf_counter() - begin
         if verbose:
             print(f"[run_ort_fusion] failed in {duration} for model_type={model_type!r}")
         return {