Google Git
Sign in
opensecura / 3p / openxla / iree / 7016b8c82b0f28c9688d27c0032349bbce2cd2cf
commit7016b8c82b0f28c9688d27c0032349bbce2cd2cf[log] [tgz]
authorTrevor Morris <tmorris@nvidia.com>Wed May 17 10:47:44 2023 -0700
committerGitHub <noreply@github.com>Wed May 17 10:47:44 2023 -0700
tree652158ecdd8babef7457a0a38767f7833d632e66
parent0e90e29fd0ba91100df0a704e9999fe82060c5d6 [diff]
Support mhlo.collective_permute with NCCL (#13502)

Adds support for `mhlo.collective_permute`.

During lowering, the `source_target_pairs` are converted into a table of
send/recv ids which can be indexed by the local rank. The send/recv id
pair is stored in the `param` field of HAL `CollectiveKind::SendRecv`.
The send and receive ids are packed as 16 bit integers into the 32 bit
param.

Fixes https://github.com/openxla/iree/issues/13100
13 files changed
tree: 652158ecdd8babef7457a0a38767f7833d632e66
  1. .devcontainer/
  2. .github/
  3. benchmarks/
  4. build_tools/
  5. compiler/
  6. docs/
  7. experimental/
  8. integrations/
  9. lib/
  10. llvm-external-projects/
  11. runtime/
  12. samples/
  13. tests/
  14. third_party/
  15. tools/
  16. .bazel_to_cmake.cfg.py
  17. .bazelignore
  18. .bazelrc
  19. .bazelversion
  20. .clang-format
  21. .dockerignore
  22. .gitignore
  23. .gitmodules
  24. .pylintrc
  25. .style.yapf
  26. .yamllint.yml
  27. AUTHORS
  28. BUILD.bazel
  29. CITATION.cff
  30. CMakeLists.txt
  31. configure_bazel.py
  32. CONTRIBUTING.md
  33. LICENSE
  34. README.md
  35. WORKSPACE
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.

CI Status

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.

Architecture Overview

IREE Architecture IREE Architecture

See our website for more information.

Presentations and Talks

  • 2021-06-09: IREE Runtime Design Tech Talk (recording and slides)
  • 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.