Skip to content

Conversation

@copybara-service
Copy link

@copybara-service copybara-service bot commented Oct 31, 2025

[JAX SC] Continuation of cl/836767096. Templatize kCreateBuckets for compile-time optimization.

  • 10.89% geomean reduction in wall time with 9.02% CPU time decrease.

@copybara-service copybara-service bot changed the title Refactor input preprocessing to optimize non-minibatching path. Remove bucketetization overhead when minibatching is disabled or in pass 1. [JAX SC] Refactor input preprocessing to optimize non-minibatching path. Remove bucketetization overhead when minibatching is disabled or in pass 1. Nov 3, 2025
@copybara-service copybara-service bot force-pushed the test_826579723 branch 2 times, most recently from 4174edd to d20eef3 Compare November 3, 2025 16:27
@copybara-service copybara-service bot changed the title [JAX SC] Refactor input preprocessing to optimize non-minibatching path. Remove bucketetization overhead when minibatching is disabled or in pass 1. [JAX SC] Continuation of cl/836767096. Templatize kCreateBuckets for compile-time optimization. Nov 25, 2025
@copybara-service copybara-service bot force-pushed the test_826579723 branch 4 times, most recently from ab8b710 to 30e0931 Compare November 26, 2025 05:22
…r compile-time optimization.

* `10.89%` geomean reduction in wall time with `9.02%` CPU time decrease.

PiperOrigin-RevId: 836943854
@copybara-service copybara-service bot merged commit 3ddc44d into main Nov 26, 2025
@copybara-service copybara-service bot deleted the test_826579723 branch November 26, 2025 06:05
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant