fix list comprehension

nod-ai · Oct 8, 2024 · 421d1b1 · 421d1b1
1 parent 5e1c1d9
commit 421d1b1
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/attentionbench/attention_utils.py b/attentionbench/attention_utils.py
@@ -70,7 +70,7 @@ def get_lowering_config(self) -> str:
             f"#iree_gpu.lowering_config<"
             + "{ "
             + f"workgroup = [{",".join([str(x) for x in self.wg_tiles])}], "
-            + f"reduction = [{",".join(str(x) for x in self.red_tiles)}]"
+            + f"reduction = [{",".join([str(x) for x in self.red_tiles])}]"
             + " }"
             + f">"
         )