From 0b75bd4000cad9253cc18b595e325a6a6ad3e46b Mon Sep 17 00:00:00 2001 From: Han-Chung Wang Date: Thu, 11 Jul 2024 11:28:33 -0700 Subject: [PATCH] [GPU] Fix the propagation control function logic. (#17869) This is a follow-up from the integrate comment: https://github.com/iree-org/iree/pull/17827#discussion_r1671197583 The comment was not addressed in the integrate PR because the CI was not stable. Some runners were off, and we decided to land the PR and sent a follow-up later. See https://discord.com/channels/689900678990135345/1080178290188374049/1260330876651307090 for more details. --- .../Codegen/Dialect/GPU/Transforms/PackToIntrinsics.cpp | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/compiler/src/iree/compiler/Codegen/Dialect/GPU/Transforms/PackToIntrinsics.cpp b/compiler/src/iree/compiler/Codegen/Dialect/GPU/Transforms/PackToIntrinsics.cpp index 4e9b40ec35b3..df0ff1c73ecc 100644 --- a/compiler/src/iree/compiler/Codegen/Dialect/GPU/Transforms/PackToIntrinsics.cpp +++ b/compiler/src/iree/compiler/Codegen/Dialect/GPU/Transforms/PackToIntrinsics.cpp @@ -100,7 +100,7 @@ void PackToIntrinsicsPass::runOnOperation() { linalg::ControlPropagationFn control = [](OpOperand *opOperand) -> bool { Operation *producer = opOperand->get().getDefiningOp(); Operation *consumer = opOperand->getOwner(); - return !getLoweringConfig(producer) || !getLoweringConfig(consumer); + return !getLoweringConfig(producer) && !getLoweringConfig(consumer); }; linalg::populateDataLayoutPropagationPatterns(patterns, control);