From aa242556ab9558b7c642dade22b4f3cd07f06f69 Mon Sep 17 00:00:00 2001 From: saienduri <77521230+saienduri@users.noreply.github.com> Date: Sun, 13 Oct 2024 23:57:12 -0700 Subject: [PATCH] fix " usage --- attentionbench/attention_utils.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) 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">" )