Vincent Wang 1bfc26685b ATen Op Supports Int Return Type and CPU Tensor Arguments (#19773) ... This PR: - add support for int as return type, will create a CPU scalar tensor for it. - add attributes to specify which arguments or returns are CPU tensors. - adjust ATen efficient attn to match latest PyTorch native function. - a Triton codegen bugfix by the way.		2024-03-06 10:11:46 +08:00
..
amp	[Better Engineering] Bump ruff to 0.0.278 and fix new lint errors (#16789)	2023-07-21 12:53:41 -07:00
api	Introduce a Nominal Checkpoint for On-Device Training (#19232)	2024-01-30 22:11:25 -08:00
experimental	Manage ORTModule configurations consistently (#16396)	2023-06-27 19:19:36 +08:00
onnxblock	Introduce a Nominal Checkpoint for On-Device Training (#19232)	2024-01-30 22:11:25 -08:00
optim	FP16 optimizer automatically detect DeepSpeed compatibility (#18084)	2023-10-25 15:11:02 +08:00
ort_triton	ATen Op Supports Int Return Type and CPU Tensor Arguments (#19773)	2024-03-06 10:11:46 +08:00
ortmodule	ATen Op Supports Int Return Type and CPU Tensor Arguments (#19773)	2024-03-06 10:11:46 +08:00
utils	Improve memory matrix for ORTModule (#19620)	2024-02-28 15:57:05 +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	Introducing customizable input names for loss in generate_artifacts. (#19705)	2024-02-29 13:40:56 -08:00