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
..
training	ATen Op Supports Int Return Type and CPU Tensor Arguments (#19773)	2024-03-06 10:11:46 +08:00
orttraining_pybind_common.h	Re-work global objects dependancies in pybind layer. (#14941)	2023-03-10 13:55:31 -08:00
orttraining_pybind_state.cc	Introduce a Nominal Checkpoint for On-Device Training (#19232)	2024-01-30 22:11:25 -08:00
orttraining_python_module.cc	[CUDA][ROCm][Training] Fix cuda/rocm provider info hash (#19398)	2024-02-05 14:35:57 -08:00
orttraining_python_module_eager.h	Run clang-format in CI (#15524)	2023-04-18 09:26:58 -07:00
pt_patch.py	Adopt linrtunner as the linting tool - take 2 (#15085)	2023-03-24 15:29:03 -07:00