Vincent Wang c07a3b869c Triton Codegen for ORTModule (#15831) ... Fuse connected elementwise and reduce Ops to TritonOp and codegen triton code to run the kernel. This PR is co-edited by @wejoncy and @er3x3		2023-07-13 18:17:58 +08:00
..
experimental	Manage ORTModule configurations consistently (#16396)	2023-06-27 19:19:36 +08:00
torch_cpp_extensions	Run clang-format in CI (#15524)	2023-04-18 09:26:58 -07:00
__init__.py	Triton Codegen for ORTModule (#15831)	2023-07-13 18:17:58 +08:00
_custom_autograd_function.py	Manage ORTModule configurations consistently (#16396)	2023-06-27 19:19:36 +08:00
_custom_autograd_function_exporter.py	Consolidate ORTModule logging (#16078)	2023-06-01 10:09:12 +08:00
_custom_autograd_function_runner.py	Adopt linrtunner as the linting tool - take 2 (#15085)	2023-03-24 15:29:03 -07:00
_custom_gradient_registry.py	Adopt linrtunner as the linting tool - take 2 (#15085)	2023-03-24 15:29:03 -07:00
_custom_op_symbolic_registry.py	Manage ORTModule configurations consistently (#16396)	2023-06-27 19:19:36 +08:00
_execution_agent.py	Adopt linrtunner as the linting tool - take 2 (#15085)	2023-03-24 15:29:03 -07:00
_fallback.py	Consolidate ORTModule logging (#16078)	2023-06-01 10:09:12 +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	Triton Codegen for ORTModule (#15831)	2023-07-13 18:17:58 +08:00
_graph_execution_manager_factory.py	Manage ORTModule configurations consistently (#16396)	2023-06-27 19:19:36 +08:00
_inference_manager.py	Triton Codegen for ORTModule (#15831)	2023-07-13 18:17:58 +08:00
_io.py	Consolidate ORTModule logging (#16078)	2023-06-01 10:09:12 +08:00
_logger.py	Log ORTModule initialization overhead (#16529)	2023-07-11 14:11:29 +08:00
_onnx_models.py	Type hint for ORTModule (#15938)	2023-05-25 09:28:20 +08:00
_runtime_inspector.py	Introduce memory observer for ORTModule (#16213)	2023-06-15 15:45:36 +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	Manage ORTModule configurations consistently (#16396)	2023-06-27 19:19:36 +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	Triton Codegen for ORTModule (#15831)	2023-07-13 18:17:58 +08:00
_utils.py	Triton Codegen for ORTModule (#15831)	2023-07-13 18:17:58 +08:00
graph_transformer_registry.py	Triton Codegen for ORTModule (#15831)	2023-07-13 18:17:58 +08:00
options.py	Triton Codegen for ORTModule (#15831)	2023-07-13 18:17:58 +08:00
ortmodule.py	Manage ORTModule configurations consistently (#16396)	2023-06-27 19:19:36 +08:00