diff --git a/cpp/src/io/parquet/chunk_dict.cu b/cpp/src/io/parquet/chunk_dict.cu index aca9eed1c39..a4d9673e8fa 100644 --- a/cpp/src/io/parquet/chunk_dict.cu +++ b/cpp/src/io/parquet/chunk_dict.cu @@ -18,7 +18,7 @@ #include #include -#include +#include #include @@ -53,7 +53,8 @@ struct equality_functor { __device__ bool operator()(size_type lhs_idx, size_type rhs_idx) { // We don't call this for nulls so this is fine - return equality_compare(col.element(lhs_idx), col.element(rhs_idx)); + auto const equal = cudf::experimental::row::equality::nan_equal_physical_equality_comparator{}; + return equal(col.element(lhs_idx), col.element(rhs_idx)); } };