saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-07-04 04:07:22 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions
onnxruntime/orttraining/orttraining/python/training/ortmodule
History
pengwa 998517b209
Minor fixes (#18949) 
### Description
<!-- Describe your changes. -->



### Motivation and Context
<!-- - Why is this change required? What problem does it solve?
- If it fixes an open issue, please link to the issue here. -->
2023-12-28 20:01:06 +08:00
..
experimental Disable PERF* rules in ruff to allow better readability (#16834) 2023-07-25 15:38:22 -07:00
graph_optimizers [ORTModule] Adjust Attention Patterns for Efficient Attention ATen Fallback (#18471) 2023-11-22 15:24:05 +08:00
torch_cpp_extensions Minor fixes (#18949) 2023-12-28 20:01:06 +08:00
__init__.py [ORTModule] ATen Efficient Attention and Triton Flash Attention (#17959) 2023-10-27 10:29:27 +08:00
_custom_autograd_function.py Improve perf for stage3 training (#18099) 2023-12-15 13:32:19 +08:00
_custom_autograd_function_exporter.py Improve perf for stage3 training (#18099) 2023-12-15 13:32:19 +08:00
_custom_gradient_registry.py ResizeGrad CUDA/ROCM kernel implementation (#17772) 2023-10-20 11:39:57 -07:00
_custom_op_symbolic_registry.py Optimize 4bit Qlora training (#18131) 2023-11-02 09:46:11 -07:00
_execution_agent.py Memory optimization refactor and refinement (#17481) 2023-11-23 11:39:00 +08:00
_fallback.py Bump linter versions (#18341) 2023-11-08 13:04:40 -08:00
_fallback_exceptions.py Adopt linrtunner as the linting tool - take 2 (#15085) 2023-03-24 15:29:03 -07:00
_gradient_accumulation_manager.py Adopt linrtunner as the linting tool - take 2 (#15085) 2023-03-24 15:29:03 -07:00
_graph_execution_interface.py Adopt linrtunner as the linting tool - take 2 (#15085) 2023-03-24 15:29:03 -07:00
_graph_execution_manager.py Allow layer-wise recompute (#18566) 2023-12-12 08:44:05 +08:00
_graph_execution_manager_factory.py ORTModule log clean up (#16795) 2023-07-26 12:42:50 +08:00
_inference_manager.py Tune ORTModule logging experience a bit (#18298) 2023-11-08 17:42:50 +08:00
_io.py Skip module clone for preparing large model export (#18663) 2023-12-05 12:41:17 -08:00
_logger.py Memory optimization refactor and refinement (#17481) 2023-11-23 11:39:00 +08:00
_onnx_models.py Allow layer-wise recompute (#18566) 2023-12-12 08:44:05 +08:00
_runtime_inspector.py Allow layer-wise recompute (#18566) 2023-12-12 08:44:05 +08:00
_torch_module_factory.py Manage ORTModule configurations consistently (#16396) 2023-06-27 19:19:36 +08:00
_torch_module_interface.py Adopt linrtunner as the linting tool - take 2 (#15085) 2023-03-24 15:29:03 -07:00
_torch_module_ort.py ORTModule log clean up (#16795) 2023-07-26 12:42:50 +08:00
_torch_module_pytorch.py Adopt linrtunner as the linting tool - take 2 (#15085) 2023-03-24 15:29:03 -07:00
_training_manager.py Allow layer-wise recompute (#18566) 2023-12-12 08:44:05 +08:00
_utils.py Use full qualified name for PythonOp export (#17021) 2023-08-09 10:58:33 +08:00
_zero_stage3_compatibility.py Improve perf for stage3 training (#18099) 2023-12-15 13:32:19 +08:00
graph_optimizer_registry.py [ORTModule] ATen Efficient Attention and Triton Flash Attention (#17959) 2023-10-27 10:29:27 +08:00
options.py Allow layer-wise recompute (#18566) 2023-12-12 08:44:05 +08:00
ortmodule.py ORTModule log clean up (#16795) 2023-07-26 12:42:50 +08:00