YQL-15415 Push down Unordered.

1 files changed, 9 insertions, 0 deletions

diff --git a/ydb/library/yql/core/common_opt/yql_co_flow2.cpp b/ydb/library/yql/core/common_opt/yql_co_flow2.cpp
index 4d1193a6f6..9919845b3a 100644
--- a/ydb/library/yql/core/common_opt/yql_co_flow2.cpp
+++ b/ydb/library/yql/core/common_opt/yql_co_flow2.cpp
@@ -2180,6 +2180,15 @@ void RegisterCoFlowCallables2(TCallableOptimizerMap& map) {
             return ctx.ChangeChildren(node->Head(), std::move(children));
         }
 
+        if (node->Head().IsCallable({
+            "Collect", "LazyList", "ForwardList","Iterator","FromFlow","ToFlow","AssumeUnique","AssumeDistinct",
+            "FilterNullMembers","SkipNullMembers","FilterNullElements","SkipNullElements",
+            "ExpandMap","WideMap","WideFilter","NarrowMap","NarrowFlatMap","NarrowMultiMap"
+        })) {
+            YQL_CLOG(DEBUG, Core) << "Swap " << node->Content() << " with " << node->Head().Content();
+            return ctx.SwapWithHead(*node);
+        }
+
         if (node->Head().IsCallable("TopSort")) {
             YQL_CLOG(DEBUG, Core) << node->Content() << " over " << node->Head().Content();
             return ctx.RenameNode(node->Head(), "Top");