Skip to content

Support setting max_rows_per_partition and report total time in pdsh benchmarks #19158

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
wants to merge 2 commits into
base: branch-25.08
Choose a base branch
from

Conversation

Matt711
Copy link
Contributor

@Matt711 Matt711 commented Jun 13, 2025

Description

Support setting max_rows_per_partition and report total mean time in pdsh benchmarks

Checklist

  • I am familiar with the Contributing Guidelines.
  • New or existing tests cover these changes.
  • The documentation is up to date with these changes.

@Matt711 Matt711 requested a review from a team as a code owner June 13, 2025 14:32
@Matt711 Matt711 requested review from wence- and mroeschke June 13, 2025 14:32
@Matt711 Matt711 added improvement Improvement / enhancement to an existing function non-breaking Non-breaking change labels Jun 13, 2025
@github-actions github-actions bot added Python Affects Python cuDF API. cudf-polars Issues specific to cudf-polars labels Jun 13, 2025
@GPUtester GPUtester moved this to In Progress in cuDF Python Jun 13, 2025
@@ -196,6 +198,12 @@ def summarize(self) -> None:
f"mean time: {np.mean([record.duration for record in records]):0.4f}"
)
print("=======================================")
total_mean_time = sum(
np.mean([record.duration for record in records])
Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

You can use statistics.mean from the standard library to avoid the numpy import

@@ -196,6 +198,12 @@ def summarize(self) -> None:
f"mean time: {np.mean([record.duration for record in records]):0.4f}"
Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Similar comment here

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
cudf-polars Issues specific to cudf-polars improvement Improvement / enhancement to an existing function non-breaking Non-breaking change Python Affects Python cuDF API.
Projects
Status: In Progress
Development

Successfully merging this pull request may close these issues.

2 participants