diff --git a/attentionbench/attention_utils.py b/attentionbench/attention_utils.py index cc9793d..863755d 100644 --- a/attentionbench/attention_utils.py +++ b/attentionbench/attention_utils.py @@ -69,8 +69,8 @@ def get_lowering_config(self) -> str: return ( 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"workgroup = [{','.join([str(x) for x in self.wg_tiles])}], " + + f"reduction = [{','.join([str(x) for x in self.red_tiles])}]" + " }" + f">" )