attribute change

nod-ai · Jan 14, 2025 · dcedcbd · dcedcbd
1 parent f8d6ecd
commit dcedcbd
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/attentionbench/attention_utils.py b/attentionbench/attention_utils.py
@@ -93,7 +93,7 @@ def get_translation_info(self) -> str:
             llvm_func_attrs += [f'"denormal-fp-math-f32" = "preserve-sign"']
         return (
             f"#iree_codegen.translation_info<"
-            + f"LLVMGPUVectorDistribute"
+            + f"pipeline = LLVMGPUVectorDistribute"
             + f" workgroup_size = [{self.N_warp * self.M_warp * 64}]"
             + f" subgroup_size = 64"
             + f" ,{{mma_schedule = {self.get_mma_schedule()}"