onnxruntime/orttraining/orttraining/python/training/ortmodule
Baiju Meswani 8b98ecad70
Change RuntimeError to ImportError (#17380)
The `onnxruntime-validation` for ORTModule checks for `ImportError`:


44101e8771/onnxruntime/python/onnxruntime_validation.py (L73-L75)

If any other kind of error is raised, it does not silently fail and will
raise an exception. This causes a problem when ortmodule is explicitly
not made available on win/mac packages since we currently raise a
RuntimeError.

Resolves issue:
https://github.com/microsoft/onnxruntime-training-examples/issues/161
2023-09-01 09:56:40 +08:00
..
experimental
torch_cpp_extensions PythonOp Enhancement: Bool and Tuple[Bool] Constants, Materialize Grads, Empty Inputs, Save In Context (#16828) 2023-08-15 13:31:04 +08:00
__init__.py Change RuntimeError to ImportError (#17380) 2023-09-01 09:56:40 +08:00
_custom_autograd_function.py
_custom_autograd_function_exporter.py Introduce ZeROOffloadSubscriber for ORTModule (#17006) 2023-08-25 00:15:22 +08:00
_custom_autograd_function_runner.py Introduce ZeROOffloadSubscriber for ORTModule (#17006) 2023-08-25 00:15:22 +08:00
_custom_gradient_registry.py
_custom_op_symbolic_registry.py Introduce ZeROOffloadSubscriber for ORTModule (#17006) 2023-08-25 00:15:22 +08:00
_execution_agent.py
_fallback.py
_fallback_exceptions.py
_gradient_accumulation_manager.py
_graph_execution_interface.py
_graph_execution_manager.py Introduce ZeROOffloadSubscriber for ORTModule (#17006) 2023-08-25 00:15:22 +08:00
_graph_execution_manager_factory.py
_inference_manager.py Refactor schema extraction and output unflattening (#16894) 2023-08-04 13:58:21 +08:00
_io.py Introduce ZeROOffloadSubscriber for ORTModule (#17006) 2023-08-25 00:15:22 +08:00
_logger.py Fix few small bugs (#17019) 2023-08-07 14:01:36 +08:00
_onnx_models.py Save optimized pre_grad graph once ready (#16816) 2023-08-02 14:05:26 +08:00
_runtime_inspector.py
_torch_module_factory.py
_torch_module_interface.py
_torch_module_ort.py
_torch_module_pytorch.py
_training_manager.py Refactor schema extraction and output unflattening (#16894) 2023-08-04 13:58:21 +08:00
_utils.py Use full qualified name for PythonOp export (#17021) 2023-08-09 10:58:33 +08:00
graph_transformer_registry.py
options.py Introduce ZeROOffloadSubscriber for ORTModule (#17006) 2023-08-25 00:15:22 +08:00
ortmodule.py