onnxruntime/orttraining/orttraining/python/training
jingyanwangms 319481898c
Give a triton library missing warning instead of silently turn off (#19276)
### Description
When USE_ORTMODULE_TRITON is set to 1 but there's no triton library,
triton function is silently turned off. This adds a warning
2024-02-01 15:25:33 -08:00
..
amp
api Introduce a Nominal Checkpoint for On-Device Training (#19232) 2024-01-30 22:11:25 -08:00
experimental
onnxblock Introduce a Nominal Checkpoint for On-Device Training (#19232) 2024-01-30 22:11:25 -08:00
optim
ort_triton [ORTModule] Handle Cast on Constant Number on Triton Code-gen (#19321) 2024-01-30 17:04:01 +08:00
ortmodule Give a triton library missing warning instead of silently turn off (#19276) 2024-02-01 15:25:33 -08:00
utils ORTModule memory improvement (#18924) 2024-01-16 08:57:37 +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 Introduce a Nominal Checkpoint for On-Device Training (#19232) 2024-01-30 22:11:25 -08:00