+ optimizer

Tony-Romanov · Tony-Romanov · commit 8ea0192d5e17 · 2024-07-16T11:22:06.000+02:00
diff --git a/ydb/library/yql/core/common_opt/yql_co_flow1.cpp b/ydb/library/yql/core/common_opt/yql_co_flow1.cpp
@@ -1477,7 +1477,7 @@ TExprNode::TPtr OptimizeFlatMap(const TExprNode::TPtr& node, TExprContext& ctx,
     {
         auto canPush = [&](const auto& child) {
             // we push FlatMap over Extend only if it can later be fused with child
-            return child->IsCallable({Ordered ? "OrderedFlatMap" : "FlatMap", "GroupByKey", "CombineByKey", "PartitionByKey", "PartitionsByKeys",
+            return child->IsCallable({Ordered ? "OrderedFlatMap" : "FlatMap", "GroupByKey", "CombineByKey", "PartitionByKey", "PartitionsByKeys", "ShuffleByKeys",
                                       "ListIf", "FlatListIf", "AsList", "ToList"}) && optCtx.IsSingleUsage(*child);
         };
         if (AllOf(node->Head().ChildrenList(), canPush)) {

Original file line number	Diff line number	Diff line change
`@@ -1477,7 +1477,7 @@ TExprNode::TPtr OptimizeFlatMap(const TExprNode::TPtr& node, TExprContext& ctx,`
`1477`	`1477`	`{`
`1478`	`1478`	`auto canPush = [&](const auto& child) {`
`1479`	`1479`	`// we push FlatMap over Extend only if it can later be fused with child`
`1480`		`- return child->IsCallable({Ordered ? "OrderedFlatMap" : "FlatMap", "GroupByKey", "CombineByKey", "PartitionByKey", "PartitionsByKeys",`
	`1480`	`+ return child->IsCallable({Ordered ? "OrderedFlatMap" : "FlatMap", "GroupByKey", "CombineByKey", "PartitionByKey", "PartitionsByKeys", "ShuffleByKeys",`
`1481`	`1481`	`"ListIf", "FlatListIf", "AsList", "ToList"}) && optCtx.IsSingleUsage(*child);`
`1482`	`1482`	`};`
`1483`	`1483`	`if (AllOf(node->Head().ChildrenList(), canPush)) {`