From 421d1b11fd566d966338de51f88ce8bffd083855 Mon Sep 17 00:00:00 2001 From: Kunwar Grover Date: Tue, 8 Oct 2024 17:25:32 +0100 Subject: [PATCH] fix list comprehension --- attentionbench/attention_utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/attentionbench/attention_utils.py b/attentionbench/attention_utils.py index 7409a14..cc9793d 100644 --- 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">" )