Skip to content

Commit

Permalink
format the code
Browse files Browse the repository at this point in the history
Signed-off-by: Bangtian Liu <[email protected]>
  • Loading branch information
bangtianliu committed Feb 25, 2025
1 parent 18bfe4e commit 8c976b6
Showing 1 changed file with 3 additions and 3 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -71,15 +71,15 @@ struct StripAttentionOpCompilationInfo final

if (DictionaryAttr decompositionConfig =
attentionOp.getDecompositionConfigAttr()) {
DictionaryAttr modifiedDecompositionConfig = DictionaryAttr::get(
DictionaryAttr newConfig = DictionaryAttr::get(
decompositionConfig.getContext(),
llvm::filter_to_vector(decompositionConfig, [&](NamedAttribute attr) {
llvm::filter_to_vector(decompositionConfig, [](NamedAttribute attr) {
return attr.getName() !=
IREE::LinalgExt::AttentionOp::getQKAttrStr() &&
attr.getName() !=
IREE::LinalgExt::AttentionOp::getPVAttrStr();
}));
attentionOp.setDecompositionConfigAttr(modifiedDecompositionConfig);
attentionOp.setDecompositionConfigAttr(newConfig);
}
return success();
}
Expand Down

0 comments on commit 8c976b6

Please sign in to comment.