torch-mlir/examples/lazytensor
Sean Silva 5b6902e31c Dual license the torch-mlir project.
This commit (with approval from all contributors) dual licenses
the torch-mlir project under both the standard LLVM license and the
standard PyTorch license. This will facilitate moving code between
torch-mlir and the two upstream projects.

The standard file comment is now:

```
// This file is licensed under the Apache License v2.0 with LLVM Exceptions.
// See https://llvm.org/LICENSE.txt for license information.
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
// Also available under a BSD-style license. See LICENSE.
```

See `LICENSE` in the project root for the terms of both licenses.
2021-10-01 10:46:08 -07:00
..
README.md Add notes on missing lazy tensor ops for ResNet18 and MaskRCNN 2021-09-30 16:38:12 -07:00
builder.py Dual license the torch-mlir project. 2021-10-01 10:46:08 -07:00

README.md

Future Work for Lazy Tensor Core

In the last part of the section Understand The Metrics Report, it is mentioned that after running the metrics report,

If you see aten:: ops other than nonzero and _local_scalar_dense, that usually means a missing lowering in the accelerator plugin.

Looking at the sample output and the sample output produced by running a ResNet18 model and a MaskRCNN model, respectively, on the Lazy Tensor Core using the TorchScript backend, the following operations are needed and not yet supported by the backend:

  • aten::convolution_overrideable
  • aten::max_pool2d_with_indices
  • aten::mean.out
  • aten::sort
  • aten::arange.start_out
  • aten::bitwise_and.Tensor_out
  • aten::clamp.out
  • aten::exp.out
  • aten::index.Tensor
  • aten::nonzero
  • aten::rsqrt.out
  • aten::sigmoid.out
  • aten::topk.values
  • aten::upsample_nearest2d.out

Note: This list is incomplete because currently the MaskRCNN example crashes halfway through when run on LTC. The output error can also be found in the MaskRCNN sample output.