Bump IREE to 20231026.688 (#113)

nod-ai · Oct 27, 2023 · 58f8459 · 58f8459
1 parent 1312259
commit 58f8459
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 4 deletions.
diff --git a/.github/workflows/test.yml b/.github/workflows/test.yml
@@ -27,6 +27,12 @@ jobs:
       - name: Sync source deps
         run: |
           python -m pip install --upgrade pip
+          # Note: We install in three steps in order to satisfy requirements
+          # from non default locations first. Installing the PyTorch CPU
+          # wheels saves multiple minutes and a lot of bandwidth on runner setup.
+          pip install --index-url https://download.pytorch.org/whl/cpu \
+            -r pytorch-cpu-requirements.txt \
+            -r torchvision-requirements.txt          
           pip install --upgrade -r requirements.txt
           pip install -e  .[testing]
 

diff --git a/requirements.txt b/requirements.txt
@@ -7,5 +7,5 @@
 -r pytorch-cpu-requirements.txt
 -r torchvision-requirements.txt
 
-iree-compiler==20231025.687
-iree-runtime==20231025.687
+iree-compiler==20231026.688
+iree-runtime==20231026.688
diff --git a/tests/transforms/quantization/mm_group_quant_test.py b/tests/transforms/quantization/mm_group_quant_test.py
@@ -28,8 +28,6 @@ def setUp(self):
             Path(__file__).resolve().parent / "mm_f32_to_int4.mlir"
         ).read_text()
 
-    # Requires IREE bump.
-    @unittest.expectedFailure
     def testBasic(self):
         with Context() as context:
             module_op = Operation.parse(self.MM_F32_TO_INT4_CONTENTS)