fixes

Author: andrej

iron/applications/llama_3.2_1b/test.py

@@ -41,11 +41,11 @@ def test_llama_3_2_1b(prompt_len, num_tokens):
         shell=True,
         capture_output=True,
         text=True,
-        timeout=300,
     )
 
+    print(result.stdout)
+    print(result.stderr)
+
     assert (
         result.returncode == 0
     ), f"Command failed with return code {result.returncode}\nStderr: {result.stderr}"
-
-    print(result.stdout)

iron/operators/gemv/design.py

@@ -7,6 +7,7 @@
 from aie.extras.context import mlir_mod_ctx
 from aie.dialects.aie import *
 from aie.dialects.aiex import *
+import aie.dialects.index as index
 from aie.helpers.dialects.scf import _for as range_
 from aie.iron import Kernel, ObjectFifo, Program, Runtime, Worker
 from aie.iron.placers import SequentialPlacer