Skip to content

feat: Implement strict serial execution - #142

Open
AidenYu1673 wants to merge 3 commits into
maxtext/devfrom
maxtext/user/aiden/fix-maxtext-moe-tpu-e2e
Open

feat: Implement strict serial execution#142
AidenYu1673 wants to merge 3 commits into
maxtext/devfrom
maxtext/user/aiden/fix-maxtext-moe-tpu-e2e

Conversation

@AidenYu1673

@AidenYu1673 AidenYu1673 commented Dec 10, 2025

Copy link
Copy Markdown

Description

Enforce that the generate_gcs_location task must be completed before the entry points of the CPU/TPU Task Groups are allowed to start. This specifically addresses and prevents the nested generate_workload_id task from running prematurely.

Tests

maxtext_moe_tpu_e2e_PR3

Checklist

Before submitting this PR, please make sure (put X in square brackets):

  • I have performed a self-review of my code.
  • I have necessary comments in my code, particularly in hard-to-understand areas.
  • I have run one-shot tests and provided workload links above if applicable.
  • I have made or will make corresponding changes to the doc if needed.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants