| .. |
|
experimental
|
Disable PERF* rules in ruff to allow better readability (#16834)
|
2023-07-25 15:38:22 -07:00 |
|
graph_optimizers
|
[ORTModule] ATen Efficient Attention and Triton Flash Attention (#17959)
|
2023-10-27 10:29:27 +08:00 |
|
torch_cpp_extensions
|
Fix convergence for dolly+stage3 training (#17685)
|
2023-10-07 08:40:19 +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
|
Manage ORTModule configurations consistently (#16396)
|
2023-06-27 19:19:36 +08:00 |
|
_custom_autograd_function_exporter.py
|
Customize _get_tensor_rank for model export in stage3 (#18294)
|
2023-11-07 16:37:11 +08:00 |
|
_custom_autograd_function_runner.py
|
Avoid one time clone to save memory peak (#17934)
|
2023-10-21 19:45:45 +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
|
Adopt linrtunner as the linting tool - take 2 (#15085)
|
2023-03-24 15:29:03 -07: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
|
Tune ORTModule logging experience a bit (#18298)
|
2023-11-08 17:42:50 +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
|
Zhijxu/improve ortmodule python perf a little bit (#13716)
|
2023-11-07 15:24:57 +08:00 |
|
_logger.py
|
Tune ORTModule logging experience a bit (#18298)
|
2023-11-08 17:42:50 +08:00 |
|
_onnx_models.py
|
Save optimized pre_grad graph once ready (#16816)
|
2023-08-02 14:05:26 +08:00 |
|
_runtime_inspector.py
|
ORTModule log clean up (#16795)
|
2023-07-26 12:42:50 +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
|
Tune ORTModule logging experience a bit (#18298)
|
2023-11-08 17:42:50 +08:00 |
|
_utils.py
|
Use full qualified name for PythonOp export (#17021)
|
2023-08-09 10:58:33 +08:00 |
|
_zero_stage3_compatibility.py
|
Customize _get_tensor_rank for model export in stage3 (#18294)
|
2023-11-07 16:37:11 +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
|
Tune ORTModule logging experience a bit (#18298)
|
2023-11-08 17:42:50 +08:00 |
|
ortmodule.py
|
ORTModule log clean up (#16795)
|
2023-07-26 12:42:50 +08:00 |
