f6f890520b
This includes custom op matching for decomposed operations and fusing dequantization into dense operations. As a validation we compare to the dequant+mm torch implementation. |
||
|---|---|---|
| .. | ||
| 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 | ||
| tensor.cpp | ||
| tensor.h | ||
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 overridden 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.