Skip to content

/ iree

👻

google.github.io/iree/
Apache-2.0 License
700 stars 152 forks
Star
Notifications
main
Switch branches/tags
16 branches 297 tags
Code
Clone

Use Git or checkout with SVN using the web URL.

Latest commit

@stellaraccident
stellaraccident Introduce input_type to specify input pipeline and re-organize the (#…
804266f Jun 2, 2021
Introduce `input_type` to specify input pipeline and re-organize the (#… 
…6030)

* Introduce `input_type` to specify input pipeline and re-organize the
input phase to be self contained.

* This includes a large amount of fiddling with test flags now that we
default to no input legalization.
* Eliminates all of the v0 shape propagation logic, with the exception
of some expansions happening at func boundaries (which should be
reworked but isn't invasive like the rest of it is).
* Creates a dedicated MHLO integrations/tensorflow input pipeline which
encapsulates some of the weird that goes beyond just MHLO ops (i.e.
deals with things that come from an XLA input, which the core compiler
does not touch).
* Fully legalizes both TF and XLA at the import level, allowing the core
compiler to be invoked with no input pipeline.
* Introduces hard checks on the core compiler boundary to emit useful
errors when input dialects have not been previously legalized away.
* Switches to the CHLO direct broadcast lowerings, which allow us to
directly handle the vast majority of dynamic shapes without involving
the shape/shapex dialect.

* Disable atanh test on vulkan. See #6076
* Disable tosa pad test on vulkan. See #6077
* Add flag to manual riscv test.
804266f

Git stats

Files

Permalink
Failed to load latest commit information.
Type
Name
Latest commit message
Commit time
.github
Relicense IREE under the LLVM license (#6020)
May 26, 2021
bindings
Introduce input_type to specify input pipeline and re-organize the (#…
Jun 2, 2021
build_tools
Introduce input_type to specify input pipeline and re-organize the (#…
Jun 2, 2021
colab
Add a colab notebook demonstrating TFLite import process (#5990)
Jun 2, 2021
docs
Relicense IREE under the LLVM license (#6020)
May 26, 2021
experimental
Relicense IREE under the LLVM license (#6020)
May 26, 2021
integrations/tensorflow
Introduce input_type to specify input pipeline and re-organize the (#…
Jun 2, 2021
iree
Introduce input_type to specify input pipeline and re-organize the (#…
Jun 2, 2021
scripts
[cmake] Support defining benchmark suites from MLIR input modules (#6024
Jun 1, 2021
third_party
Synchronize submodules with LLVM at llvm/llvm-project@97d2349
Jun 1, 2021
.bazelignore
Relicense IREE under the LLVM license (#6020)
May 26, 2021
.bazelrc
Relicense IREE under the LLVM license (#6020)
May 26, 2021
.bazelversion
Upgrade Bazel to 3.7.1 (#4242)
Dec 21, 2020
.clang-format
Relicense IREE under the LLVM license (#6020)
May 26, 2021
.gitignore
Prep for new website: move existing docs and disable current publicat…
May 5, 2021
.gitmodules
Removing unused pffft submodule. (#5916)
May 15, 2021
.style.yapf
Add yapf style config for automatic python formatting.
Jul 21, 2020
.yamllint.yml
Relicense IREE under the LLVM license (#6020)
May 26, 2021
AUTHORS
Add Nod to AUTHORS (#6028)
May 26, 2021
BUILD.bazel
Relicense IREE under the LLVM license (#6020)
May 26, 2021
CMakeLists.txt
[cmake] Support defining benchmark suites from MLIR input modules (#6024
Jun 1, 2021
CONTRIBUTING.md
Relicense IREE under the LLVM license (#6020)
May 26, 2021
LICENSE
Relicense IREE under the LLVM license (#6020)
May 26, 2021
README.md
Relicense IREE under the LLVM license (#6020)
May 26, 2021
SUBMODULE_VERSIONS.txt
Synchronize submodules with LLVM at llvm/llvm-project@97d2349
Jun 1, 2021
WORKSPACE
Relicense IREE under the LLVM license (#6020)
May 26, 2021
configure_bazel.py
Relicense IREE under the LLVM license (#6020)
May 26, 2021
IREE: Intermediate Representation Execution Environment Project Status Communication Channels Related Project Channels Build Status Presentations and Talks License

README.md

IREE: Intermediate Representation Execution Environment

IREE (Intermediate Representation Execution Environment, pronounced as "eerie") is an MLIR-based end-to-end compiler and runtime that lowers Machine Learning (ML) models to a unified IR that scales up to meet the needs of the datacenter and down to satisfy the constraints and special considerations of mobile and edge deployments.

See our website for project details, user guides, and instructions on building from source.

Project Status

IREE is still in its early phase. We have settled down on the overarching infrastructure and are actively improving various software components as well as project logistics. It is still quite far from ready for everyday use and is made available without any support at the moment. With that said, we welcome any kind of feedback on any communication channels!

Communication Channels

Related Project Channels

  • MLIR topic within LLVM Discourse: IREE is enabled by and heavily relies on MLIR. IREE sometimes is referred to in certain MLIR discussions. Useful if you are also interested in MLIR evolution.

Build Status

CI System Build System Platform Architecture Component Status
Kokoro Bazel Linux x86 Core kokoro_status_bazel_linux_x86_core
Kokoro CMake & Bazel Linux x86-swiftshader Integrations kokoro_status_cmake-bazel_linux_x86-swiftshader_integrations
Kokoro CMake & Bazel Linux x86-turing Integrations kokoro_status_cmake-bazel_linux_x86-turing_integrations
Kokoro CMake Linux x86-swiftshader Core + Bindings kokoro_status_cmake_linux_x86-swiftshader
Kokoro CMake Linux x86-swiftshader-asan Core + Bindings kokoro_status_cmake_linux_x86-swiftshader-asan
Kokoro CMake Linux x86-turing Core + Bindings kokoro_status_cmake_linux_x86-turing
Kokoro CMake Android arm64-v8a Runtime (build only) kokoro_status_cmake_android_arm64-v8a
BuildKite CMake Android arm64-v8a Runtime buildkite-status-cmake-android-arm

Presentations and Talks

  • 2020-08-20: IREE CodeGen: MLIR Open Design Meeting Presentation (recording and slides)
  • 2020-03-18: Interactive HAL IR Walkthrough (recording)
  • 2020-01-31: End-to-end MLIR Workflow in IREE: MLIR Open Design Meeting Presentation (recording and slides)

License

IREE is licensed under the terms of the Apache 2.0 License with LLVM Exceptions. See LICENSE for more information.

About

👻

google.github.io/iree/

Topics

tensorflow vulkan spirv mlir

Resources

Readme

License

Apache-2.0 License

Releases

297 tags

Packages

No packages published

Contributors 75

+ 64 contributors

Languages