Skip to content

Add max_inflight_computations config parameter and clean up getattr lookups - #4845

Open
copybara-service[bot] wants to merge 1 commit into
mainfrom
test_963028167
Open

Add max_inflight_computations config parameter and clean up getattr lookups#4845
copybara-service[bot] wants to merge 1 commit into
mainfrom
test_963028167

Conversation

@copybara-service

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

Copy link
Copy Markdown
Contributor

Add max_inflight_computations config parameter and clean up getattr lookups

  • Add max_inflight_computations config parameter to base.yml and types.py.
  • Replace defensive getattr() lookups with typed config attribute access across checkpointing, metrics, inflight_throttler, gradient_accumulation, and maxtext_engine.
  • Clean up compare_training_engine test harness to use default model configuration, make_dummy_data, and dynamic learning rate schedule.

@codecov

codecov Bot commented Aug 11, 2026

Copy link
Copy Markdown

Codecov Report

❌ Patch coverage is 33.33333% with 4 lines in your changes missing coverage. Please review.

Files with missing lines Patch % Lines
src/maxtext/training_engine/maxtext_engine.py 25.00% 0 Missing and 3 partials ⚠️
src/maxtext/training_engine/metrics.py 0.00% 0 Missing and 1 partial ⚠️

📢 Thoughts on this report? Let us know!

@copybara-service
copybara-service Bot force-pushed the test_963028167 branch 3 times, most recently from a43d2ea to eeeda5a Compare August 13, 2026 02:27
…ookups

- Add max_inflight_computations config parameter to base.yml and types.py.
- Replace defensive getattr() lookups with typed config attribute access across checkpointing, metrics, inflight_throttler, gradient_accumulation, and maxtext_engine.
- Clean up compare_training_engine test harness to use default model configuration, make_dummy_data, and dynamic learning rate schedule.

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