Skip to content

Prefill return first token#105

Merged
jwyang-google merged 16 commits into
mainfrom
prefill-return-first-token
Jun 28, 2024
Merged

Prefill return first token#105
jwyang-google merged 16 commits into
mainfrom
prefill-return-first-token

Conversation

@jwyang-google

@jwyang-google jwyang-google commented Jun 25, 2024

Copy link
Copy Markdown
Collaborator

Modify Jetstream to make prefill return first token. Pending testing with MLPerf loadgen.

Comment thread jetstream/core/orchestrator.py

@JoeZijunZhou JoeZijunZhou left a comment

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

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

Need to update unit tests and resolve the CICD failures after all.

@JoeZijunZhou JoeZijunZhou left a comment

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

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

LGTM! Thanks Junwei!

@jwyang-google jwyang-google merged commit c3fe3ce into main Jun 28, 2024
@jwyang-google jwyang-google deleted the prefill-return-first-token branch June 28, 2024 15:42
vivianrwu pushed a commit to vivianrwu/JetStream that referenced this pull request Jul 2, 2024
Change prefill API to return first token.
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.

2 participants