Skip to content

Commit

Permalink
Expose mixed and conditional joins in pylibcudf
Browse files Browse the repository at this point in the history
  • Loading branch information
wence- committed Oct 31, 2024
1 parent e512258 commit 7b6141d
Show file tree
Hide file tree
Showing 4 changed files with 744 additions and 4 deletions.
76 changes: 76 additions & 0 deletions python/pylibcudf/pylibcudf/join.pxd
Original file line number Diff line number Diff line change
Expand Up @@ -3,6 +3,7 @@
from pylibcudf.libcudf.types cimport null_equality

from .column cimport Column
from .expressions cimport Expression
from .table cimport Table


Expand Down Expand Up @@ -37,3 +38,78 @@ cpdef Column left_anti_join(
)

cpdef Table cross_join(Table left, Table right)

cpdef tuple conditional_inner_join(
Table left,
Table right,
Expression binary_predicate,
)

cpdef tuple conditional_left_join(
Table left,
Table right,
Expression binary_predicate,
)

cpdef tuple conditional_full_join(
Table left,
Table right,
Expression binary_predicate,
)

cpdef Column conditional_left_semi_join(
Table left,
Table right,
Expression binary_predicate,
)

cpdef Column conditional_left_anti_join(
Table left,
Table right,
Expression binary_predicate,
)

cpdef tuple mixed_inner_join(
Table left_keys,
Table right_keys,
Table left_conditional,
Table right_conditional,
Expression binary_predicate,
null_equality nulls_equal
)

cpdef tuple mixed_left_join(
Table left_keys,
Table right_keys,
Table left_conditional,
Table right_conditional,
Expression binary_predicate,
null_equality nulls_equal
)

cpdef tuple mixed_full_join(
Table left_keys,
Table right_keys,
Table left_conditional,
Table right_conditional,
Expression binary_predicate,
null_equality nulls_equal
)

cpdef Column mixed_left_semi_join(
Table left_keys,
Table right_keys,
Table left_conditional,
Table right_conditional,
Expression binary_predicate,
null_equality nulls_equal
)

cpdef Column mixed_left_anti_join(
Table left_keys,
Table right_keys,
Table left_conditional,
Table right_conditional,
Expression binary_predicate,
null_equality nulls_equal
)
Loading

0 comments on commit 7b6141d

Please sign in to comment.