From 8f6e493c1cc8fba041a4e2d25b9f14aa892ad92f Mon Sep 17 00:00:00 2001 From: Henry Tu Date: Thu, 30 Jun 2022 15:30:35 -0400 Subject: [PATCH] Update README.md --- README.md | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/README.md b/README.md index 260636e0e4e4c..0c37c5012ef3f 100644 --- a/README.md +++ b/README.md @@ -73,8 +73,9 @@ torch-mlir prediction ### LazyTensorCore -The LazyTensorCore integration is still in progress, and is being built on the -[`torch_mlir_ltc_backend` branch](https://github.com/llvm/torch-mlir/tree/torch_mlir_ltc_backend). +Lazy Tensor Core support is provided through an abstract [`TorchMlirBackendImpl` class](python/torch_mlir/csrc/base_lazy_backend/backend_impl.h). An example implementation is avalable [here](examples/ltc_backend/ltc_backend). + +There are also examples of a [Huggingface BERT](torch_mlir_ltc_backend/examples/ltc_backend_bert.py) and [MNIST model](torch_mlir_ltc_backend/examples/ltc_backend_mnist.py) running on the example LTC backend. ### Eager Mode