6b02e9a926
* Tensor[]? support operands type support using partial codegen * aten.index.Tensor support via partial codegen * Add torch.index_put tracing support * Added optional tensor list type support for LTC/TorchMLIR lowering * Added comments Co-authored-by: Gleb Kazantaev <gleb.kazantaev@cerebras.net> |
||
|---|---|---|
| .. | ||
| ops | ||
| utils | ||
| CMakeLists.txt | ||
| README.md | ||
| backend_impl.cpp | ||
| backend_impl.h | ||
| dynamic_ir.cpp | ||
| dynamic_ir.h | ||
| ir_builder.h | ||
| mlir_lowering_context.cpp | ||
| mlir_lowering_context.h | ||
| mlir_native_functions.cpp | ||
| mlir_node.cpp | ||
| mlir_node.h | ||
| mlir_node_lowering.cpp | ||
| mlir_node_lowering.h | ||
| shape_inference.cpp | ||
README.md
Torch-MLIR Lazy Tensor Core Backend
Detailed Documentation
Detailed documentation about the architecture of this LTC backend is available here.
Summary
Contained within this directory are the components that implements the Torch-MLIR LTC backend. Note that the code style for LTC components is consistent with that of LTC itself, rather than the rest of Torch-MLIR.
The components are subclasses of the backend API interface classes found under torch/csrc/lazy/backend.
Importantly, the subclasses are still abstract classes. Pure virtual methods
such as Compile were purposefully not overriden as Torch-MLIR does not know
how to compile the model for the target hardware.
The intent is that vendor hardware specific plugins will subclass the Torch-MLIR backend classes and override the remaining pure virtual functions to complete the backend.
The Torch-MLIR LTC backend's job is to perform the lowering from ATen to MLIR. A hardware vendor's backend job is to take care of the actual compile and execution of the lowered MLIR.