fix a few things

Author: xadupre

_doc/api/helpers/helper.rst

@@ -4,4 +4,4 @@ onnx_diagnostic.helpers.helper
 
 .. automodule:: onnx_diagnostic.helpers.helper
     :no-undoc-members:
-    :exclude-members: max_diff, string_diff, string_sig, string_type
+    :exclude-members: flatten_object, max_diff, string_diff, string_sig, string_type

_doc/api/helpers/index.rst

@@ -18,6 +18,8 @@ onnx_diagnostic.helpers
     rt_helper
     torch_test_helper
 
+.. autofunction:: onnx_diagnostic.helpers.flatten_object
+
 .. autofunction:: onnx_diagnostic.helpers.max_diff
 
 .. autofunction:: onnx_diagnostic.helpers.string_diff

onnx_diagnostic/helpers/__init__.py

@@ -1 +1 @@
-from .helper import max_diff, string_diff, string_sig, string_type
+from .helper import flatten_object, max_diff, string_diff, string_sig, string_type

onnx_diagnostic/helpers/ort_session.py

@@ -101,11 +101,27 @@ def __init__(
                     providers = ["CUDAExecutionProvider", "CPUExecutionProvider"]
                 else:
                     raise ValueError(f"Unexpected value for providers={providers!r}")
-            sess = onnxruntime.InferenceSession(
-                sess if isinstance(sess, str) else sess.SerializeToString(),
-                session_options,
-                providers=providers,
-            )
+            try:
+                sess = onnxruntime.InferenceSession(
+                    sess if isinstance(sess, str) else sess.SerializeToString(),
+                    session_options,
+                    providers=providers,
+                )
+            except onnxruntime.capi.onnxruntime_pybind11_state.Fail as e:
+                if isinstance(sess, onnx.ModelProto):
+                    debug_path = "_debug_onnxruntine_evaluator_failure.onnx"
+                    onnx.save(
+                        sess,
+                        debug_path,
+                        save_as_external_data=True,
+                        all_tensors_to_one_file=True,
+                    )
+                else:
+                    debug_path = sess
+                raise RuntimeError(
+                    f"Unable to create a session stored in {debug_path!r}), "
+                    f"providers={providers}"
+                ) from e
         else:
             assert (
                 session_options is None

onnx_diagnostic/helpers/torch_test_helper.py

@@ -151,8 +151,8 @@ def forward(self, x, y):
             onnx.save(
                 proto,
                 dump_file,
-                save_as_external_data=False,
-                all_tensors_to_one_file=True,
+                save_as_external_data=True,
+                all_tensors_to_one_file=False,
             )