From bd24bb4c9b06852e1298af1b430dc88da2271aae Mon Sep 17 00:00:00 2001 From: Scott Todd Date: Mon, 29 Apr 2024 16:08:44 -0700 Subject: [PATCH 1/2] Bump IREE to latest commit. --- third_party/iree | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/third_party/iree b/third_party/iree index 31a5dcb..f2746b4 160000 --- a/third_party/iree +++ b/third_party/iree @@ -1 +1 @@ -Subproject commit 31a5dcb1910c9014db8339afd953eccaa0dc6076 +Subproject commit f2746b464fb056ddadef4315654d59f727e4c9b0 From d1ed1e36b8729cb386391fc67ccc44e3bcc428c5 Mon Sep 17 00:00:00 2001 From: Scott Todd Date: Mon, 29 Apr 2024 16:10:41 -0700 Subject: [PATCH 2/2] Pin iree-compiler pip version, update README and docs. --- .github/workflows/build-and-test.yml | 2 +- .gitmodules | 2 +- README.md | 15 ++++++++------- simple_mul.mlir | 2 +- 4 files changed, 11 insertions(+), 10 deletions(-) diff --git a/.github/workflows/build-and-test.yml b/.github/workflows/build-and-test.yml index 1f6af00..0993916 100644 --- a/.github/workflows/build-and-test.yml +++ b/.github/workflows/build-and-test.yml @@ -111,7 +111,7 @@ jobs: python-version: "3.11" - name: Install IREE compiler run: | - python -m pip install iree-compiler + python -m pip install iree-compiler==20240410.859 - name: Checkout repository uses: actions/checkout@v2 - name: Initialize submodules diff --git a/.gitmodules b/.gitmodules index c02503b..fd39cff 100644 --- a/.gitmodules +++ b/.gitmodules @@ -1,3 +1,3 @@ [submodule "third_party/iree"] path = third_party/iree - url = https://github.com/openxla/iree.git + url = https://github.com/iree-org/iree.git diff --git a/README.md b/README.md index 925bca4..7cdc336 100644 --- a/README.md +++ b/README.md @@ -20,7 +20,7 @@ If working in an existing repository then add the submodule and ensure it has its submodules initialized: ```sh -$ git submodule add https://github.com/openxla/iree.git third_party/iree/ +$ git submodule add https://github.com/iree-org/iree.git third_party/iree/ $ git submodule update --init --recursive ``` @@ -32,6 +32,7 @@ $ git \ -c submodule."third_party/llvm-project".update=none \ -c submodule."third_party/stablehlo".update=none \ -c submodule."third_party/torch-mlir".update=none \ + -c submodule."third_party/torch-mlir".update=none \ submodule update --init --recursive ``` @@ -52,11 +53,11 @@ $ cmake --build build/ --target hello_world ### Compiling the Sample Module -This sample assumes that the latest IREE compiler release is installed and used -to compile the module. For many users upgrading their `iree-compiler` install -when they bump their submodule should be sufficient to ensure the compiler and -runtime are compatible. In the future the compiler and runtime will have more -support for version shifting. +This sample assumes that a compatible IREE compiler release is installed and +used to compile the module. For many users upgrading their `iree-compiler` +install when they bump their submodule should be sufficient to ensure the +compiler and runtime are compatible. In the future the compiler and runtime +will have more support for version shifting. The sample currently assumes a CPU HAL driver and only produces a VMFB supporting that. Additional compiler options can be used to change the target @@ -67,7 +68,7 @@ can be accomplished with the `--iree-llvm-target-triple=` flag specifying the CPU architecture. ```sh -$ python -m pip install iree-compiler --upgrade --user +$ python -m pip install iree-compiler==20240410.859 --upgrade --user $ iree-compile \ --iree-hal-target-backends=llvm-cpu \ --iree-llvmcpu-target-triple=x86_64 \ diff --git a/simple_mul.mlir b/simple_mul.mlir index d8be9a9..a44bddf 100644 --- a/simple_mul.mlir +++ b/simple_mul.mlir @@ -2,7 +2,7 @@ // framework (JAX, PyTorch, TFLite, etc). // // Refer to the documentation for frontend-specific instructions: -// https://openxla.github.io/iree/guides/ml-frameworks/ +// https://iree.dev/guides/ml-frameworks/ // Simple elementwise multiply: // %result = %lhs * %rhs