Skip to content

Refactor UpdateMaxCapacities and ResizeSlabs to initialize CPU slabs before stopping all CPUs. - #603

Draft
copybara-service[bot] wants to merge 1 commit into
masterfrom
test_973865650
Draft

Refactor UpdateMaxCapacities and ResizeSlabs to initialize CPU slabs before stopping all CPUs.#603
copybara-service[bot] wants to merge 1 commit into
masterfrom
test_973865650

Conversation

@copybara-service

@copybara-service copybara-service Bot commented Aug 31, 2026

Copy link
Copy Markdown

Refactor UpdateMaxCapacities and ResizeSlabs to initialize CPU slabs before stopping all CPUs.

This avoids page faulting in the newly allocated slabs while CPUs are stopped. Since updating capacities still needs to happen while CPUs are stopped, the capacity callback needs to return the new capacities rather than the current ones. This way initializing the CPU slabs can happen before update_capacity is called.

@google-cla

google-cla Bot commented Aug 31, 2026

Copy link
Copy Markdown

Thanks for your pull request! It looks like this may be your first contribution to a Google open source project. Before we can look at your pull request, you'll need to sign a Contributor License Agreement (CLA).

View this failed invocation of the CLA check for more information.

For the most up to date status, view the checks section at the bottom of the pull request.

@copybara-service
copybara-service Bot force-pushed the test_973865650 branch 3 times, most recently from 0788a63 to aa1044d Compare August 31, 2026 18:11
@copybara-service copybara-service Bot changed the title Refactor UpdateMaxCapacities to initialize CPU slabs before stopping all CPUs. Refactor UpdateMaxCapacities and ResizeSlabs to initialize CPU slabs before stopping all CPUs. Aug 31, 2026
@copybara-service
copybara-service Bot force-pushed the test_973865650 branch 2 times, most recently from ca18624 to d139512 Compare August 31, 2026 20:19
…before stopping all CPUs.

This avoids page faulting in the newly allocated slabs while CPUs are stopped. Since updating capacities still needs to happen while CPUs are stopped, the capacity callback needs to return the new capacities rather than the current ones. This way initializing the CPU slabs can happen before update_capacity is called.

PiperOrigin-RevId: 973865650
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