Fixes cpu flag update for stateless llama from iree bump

IanNod · IanNod · commit 0c695635222a · 2023-12-07T16:39:50.000Z
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-vmvx-enable-microkernels")
             device = "llvm-cpu"
         elif device == "vulkan":
             flags.extend(

Original file line number	Diff line number	Diff line change
`@@ -237,7 +237,7 @@ def forward(token0: torch.Tensor, *state0_flat):`
`237`	`237`	`"--iree-opt-const-expr-hoisting=False",`
`238`	`238`	`]`
`239`	`239`	`if device == "cpu":`
`240`		`- flags.append("--iree-llvmcpu-enable-microkernels")`
	`240`	`+ flags.append("--iree-vmvx-enable-microkernels")`
`241`	`241`	`device = "llvm-cpu"`
`242`	`242`	`elif device == "vulkan":`
`243`	`243`	`flags.extend(`