Fixes cpu flag update for stateless llama from iree bump (#226)

nod-ai · Dec 7, 2023 · 1981cd6 · 1981cd6
1 parent 247396c
commit 1981cd6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/turbine_models/custom_models/stateless_llama.py b/python/turbine_models/custom_models/stateless_llama.py
@@ -237,7 +237,7 @@ def forward(token0: torch.Tensor, *state0_flat):
             "--iree-opt-const-expr-hoisting=False",
         ]
         if device == "cpu":
-            flags.append("--iree-llvmcpu-enable-microkernels")
+            flags.append("--iree-llvmcpu-enable-ukernels=all")
             device = "llvm-cpu"
         elif device == "vulkan":
             flags.extend(