Skip to content

Latest commit

 

History

History
 
 

base_lazy_backend

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.