Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
30 commits
Select commit Hold shift + click to select a range
03ef8ad
Add dynamic join key prefilter planning
pentschev Jun 25, 2026
09f0474
Reject boolean join prefilter key limits
pentschev Jun 26, 2026
6f95d33
Consolidate optional config converters
pentschev Jun 26, 2026
519cd1d
Remove legacy bloom prefilter threshold path
pentschev Jun 26, 2026
d1cd0a1
Merge remote-tracking branch 'upstream/main' into cudf-polars/join-pr…
pentschev Jun 26, 2026
09a744e
Fix linting
pentschev Jun 26, 2026
7da769c
Accept integer join prefilter thresholds
pentschev Jun 26, 2026
fd6ba23
Implement default in-place without additional variable
pentschev Jun 26, 2026
fbe886b
Document join prefilter key prefix limit
pentschev Jun 26, 2026
f01e62e
Use dataclass conversion for join prefilter trace
pentschev Jun 26, 2026
ffd5e8f
Remove redundant join prefilter decision state
pentschev Jun 26, 2026
6bee038
Remove unused keyless prefilter skip
pentschev Jun 26, 2026
47f2ce4
Assert matching join prefilter key counts
pentschev Jun 26, 2026
97466a3
Track only join prefilter apply side
pentschev Jun 26, 2026
543a73d
Remove unused prefilter partition trace
pentschev Jun 26, 2026
0dfc80d
Clarify dynamic join collective IDs
pentschev Jun 26, 2026
94e132c
Document actor trace extra metadata
pentschev Jun 26, 2026
918417d
Fix docstring linting
pentschev Jun 26, 2026
29b950e
Merge remote-tracking branch 'upstream/main' into cudf-polars/join-pr…
pentschev Jun 26, 2026
c75249d
Fix optional join prefilter default typing
pentschev Jun 26, 2026
42871f0
Merge remote-tracking branch 'upstream/main' into cudf-polars/join-pr…
pentschev Jun 26, 2026
0dafc3e
Merge remote-tracking branch 'upstream/main' into cudf-polars/join-pr…
pentschev Jun 29, 2026
702c6ad
Generalize default factory result typing
pentschev Jun 29, 2026
a37608d
Document join prefilter selection inputs
pentschev Jun 29, 2026
6bc7559
Simplify join prefilter decision flow
pentschev Jun 29, 2026
5ac7e79
Merge remote-tracking branch 'upstream/main' into cudf-polars/join-pr…
pentschev Jun 29, 2026
903084d
Fix missing coverage
pentschev Jun 29, 2026
4538aef
Fix one more missing coverage
pentschev Jun 29, 2026
c55aa6c
Merge remote-tracking branch 'upstream/main' into cudf-polars/join-pr…
pentschev Jun 29, 2026
d1936aa
Merge remote-tracking branch 'upstream/main' into cudf-polars/join-pr…
pentschev Jun 29, 2026
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
# SPDX-FileCopyrightText: Copyright (c) 2025-2026, NVIDIA CORPORATION & AFFILIATES.
# SPDX-FileCopyrightText: Copyright (c) 2025-2026, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
# SPDX-License-Identifier: Apache-2.0
"""Common utilities for collective operations."""

Expand Down Expand Up @@ -138,8 +138,8 @@ def __enter__(self) -> dict[IR, list[int]]:
_get_new_collective_id_unsafe(),
]
elif isinstance(node, Join) and self.dynamic_planning_enabled:
# Join needs 4 IDs: size allgather, left shuffle/bcast,
# right shuffle/bcast, bloom filter
# Join needs 4 IDs: allgather, left shuffle, right shuffle,
# and bloom filter.
self.collective_id_map[node] = [
_get_new_collective_id_unsafe(),
_get_new_collective_id_unsafe(),
Expand Down
Loading
Loading