Refresh IREE requirement to pull changes for async-exec mode. (#109)

Author: raikonenfnu

.github/workflows/test.yml

@@ -27,9 +27,7 @@ jobs:
       - name: Sync source deps
         run: |
           python -m pip install --upgrade pip
-          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]
 
       - name: Run tests

python/shark_turbine/aot/support/procedural/iree_emitter.py

@@ -313,8 +313,11 @@ def tensor_splat(
 
     @emitter
     def tensor_trace(self, key: str, *ts: BuildableTensorType):
+        dynamic_dims = []
+        for t in ts:
+            dynamic_dims.extend(t.get_only_dynamic_dim_values())
         ts = [cast_tensor_value(t).ir_value for t in ts]
-        flow_d.TensorTraceOp(StringAttr.get(key), ts)
+        flow_d.TensorTraceOp(StringAttr.get(key), ts, dynamic_dims)
 
 
 # Circular imports to resolve typing.

requirements.txt

@@ -7,5 +7,5 @@
 -r pytorch-cpu-requirements.txt
 -r torchvision-requirements.txt
 
-iree-compiler==20231004.665
-iree-runtime==20231004.665
+iree-compiler==20231025.687
+iree-runtime==20231025.687

tests/aot/iree_procedural_test.py

@@ -69,7 +69,7 @@ def foobar(self, x=AbstractTensor(None), y=AbstractTensor(3)):
         inst = BasicModule(context=Context(), import_to=None)
         module_str = str(CompiledModule.get_mlir_module(inst))
         print(module_str)
-        self.assertIn('flow.tensor.trace {key = "DEBUG"} %arg0, %arg1', module_str)
+        self.assertIn('flow.tensor.trace "DEBUG" = [%arg0 : tensor<?xf32>{%dim}, %arg1 : tensor<3xf32>]', module_str)
 
     def testStoreDynamic(self):
         class BasicModule(CompiledModule):

tests/dynamo/importer_dynamic_test.py

@@ -109,7 +109,6 @@ def forward(self, a):
 
 
 class ImportSmokeTests(unittest.TestCase):
-    @unittest.expectedFailure
     def testStaticExport(self):
         """
         'tensor.collapse_shape' op expected dimension 0 of collapsed type to be static value of 1
@@ -129,7 +128,6 @@ def testStaticExport(self):
         g, guards = f(inp=inp_example, bias=bias_example)
         g = import_compiler(g, [inp_example, bias_example])
 
-    @unittest.expectedFailure
     def testStaticExportSameSignatureTrue(self):
         """
         'tensor.collapse_shape' op expected dimension 0 of collapsed type to be static value of 1