Skip to content

Commit d324fd8

Browse files
committed
attribute change
1 parent 01fede0 commit d324fd8

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

attentionbench/attention_utils.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -93,7 +93,7 @@ def get_translation_info(self) -> str:
9393
llvm_func_attrs += [f'"denormal-fp-math-f32" = "preserve-sign"']
9494
return (
9595
f"#iree_codegen.translation_info<"
96-
+ f"LLVMGPUVectorDistribute"
96+
+ f"pipeline = LLVMGPUVectorDistribute"
9797
+ f" workgroup_size = [{self.N_warp * self.M_warp * 64}]"
9898
+ f" subgroup_size = 64"
9999
+ f" ,{{mma_schedule = {self.get_mma_schedule()}"

0 commit comments

Comments
 (0)