Gitiles
Code Review Sign In
review.mlplatform.org / tosa / tosa_mlir_translator / refs/tags/v0.90a0
tag3d3c2b5e5197cc6e95f4fb1d09a14a9309998a5e
taggerEric Kunze <eric.kunze@arm.com>Tue Oct 03 16:40:16 2023 -0700
objectedc04e0ef1a93d1bf4578308d620dd49c10e1ad4 
Start of work on TOSA v0.90
commitedc04e0ef1a93d1bf4578308d620dd49c10e1ad4[log] [tgz]
authorJerry Ge <jerry.ge@arm.com>Mon Sep 19 21:07:38 2022 -0700
committerTai Ly <tai.ly@arm.com>Thu Sep 07 13:17:16 2023 -0700
tree1044165c16528a17337f81aa15361c7b77aafe7f
parent7566d1235cb646e46531c2eb34757cb4b3efa933 [diff]
[tosa_mlir_translator] Support Tosa StatefulOps

Signed-off-by: Jerry Ge <jerry.ge@arm.com>
Change-Id: I6ce5a917cada436f6a80e6d85e670d6cd44e01e9
3 files changed
tree: 1044165c16528a17337f81aa15361c7b77aafe7f
  1. .clang-format
  2. .gitmodules
  3. .pre-commit-config.yaml
  4. CMakeLists.txt
  5. LICENSE.txt
  6. README.md
  7. include/
  8. src/
  9. third_party/
README.md

TOSA MLIR Translator

Introduction

The TOSA MLIR Translator repository implements translators between the TOSA MLIR dialect and serialized representations.

The current implementation supports serialization from MLIR form to flatbuffers. A deserializer from flatbuffers to MLIR form is in development.

Dependencies

##TOSA serialization library https://review.mlplatform.org/plugins/gitiles/tosa/serialization_lib The library includes a FlatBuffers schema and a C++ API for reading and writing a TOSA graph as a flatbuffer.

Compiling

This repository does not currently build standalone. It must be included within another MLIR repository with a pass manager registering the passes implemented within this repository.

The included CMake rules can be used to add this repository as a submodule. The include/SerializationPasses.h enables MLIR pass registration inclusion.

If target "tosa_serialize" is linked correctly, you should able to see "--tosa-serialize" and "--tosa-serialize-json" options available in your MLIR pass manager/MLIR optimizer.

Usage

To serialize a TOSA MLIR graph to TOSA flatbuffer binary file:

<YOUR_MLIR_OPTIMIZER> --tosa-serialize <TOSA_MLIR_GRAPH>
--tosa-flatbuffer-filename <TOSA_FLATBUFFER_FILENAME>

To serialize a TOSA MLIR graph to TOSA flatbuffer JSON file:

<YOUR_MLIR_OPTIMIZER> --tosa-serialize <TOSA_MLIR_GRAPH>
--tosa-flatbuffer-schema <PATH_TO_TOSA_FLATBUFFER_SCHEMA>
--tosa-flatbuffer-filename <TOSA_FLATBUFFER_FILENAME>

where <PATH_TO_TOSA_FLATBUFFER_SCHEMA> is provided within the serialization library submodule in third_party/serialization_lib/schema/tosa.fbs

License

The TOSA MLIR Translator is licensed under Apache-2.0 with LLVM Exceptions.