onnxruntime/orttraining/orttraining/python/training
Vincent Wang e1d1033131
[ORTModule] Remove Unused Arguments from Generated Triton Code (#18636)
This PR:
- Remove unused arguments from generated triton code,
- Remove unnecessary mask for symbolic shape case from generated triton
code.
- Add doc for usage of ORTMODULE_TRITON_CONFIG_FILE.
2023-11-30 18:32:36 +08:00
..
amp
api [On-Device Training] Expose Parameters through the Training API (#17364) 2023-09-25 20:03:24 -07:00
experimental
onnxblock [Linter] Bump ruff and remove pylint (#17797) 2023-10-05 21:07:33 -07:00
optim FP16 optimizer automatically detect DeepSpeed compatibility (#18084) 2023-10-25 15:11:02 +08:00
ort_triton [ORTModule] Remove Unused Arguments from Generated Triton Code (#18636) 2023-11-30 18:32:36 +08:00
ortmodule Memory optimization refactor and refinement (#17481) 2023-11-23 11:39:00 +08:00
torchdynamo
utils Memory optimization refactor and refinement (#17481) 2023-11-23 11:39:00 +08:00
__init__.py Removed all the deprecated python training code and related tests and utils (#18333) 2023-11-17 18:19:21 -08:00
_utils.py Removed all the deprecated python training code and related tests and utils (#18333) 2023-11-17 18:19:21 -08:00
artifacts.py Fix opset version of the optimizer in function generate_artifacts (#18300) 2023-11-22 09:15:11 -08:00