Skip to content

[sharktank] Update Extend attention config for shortfin #2605

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom
Open
wants to merge 5 commits into
base: main
Choose a base branch
from
Open

[sharktank] Update Extend attention config for shortfin #2605

wants to merge 5 commits into from

Conversation

@archana-ramalingam
Copy link
Collaborator

@archana-ramalingam archana-ramalingam commented Oct 27, 2025

  • Add use_extend_attention to ServiceConfig
  • Update prefill name in mlir

@github-actions
Copy link
Contributor

github-actions bot commented Oct 27, 2025 •
edited
Loading

Coverage report

Click to see where and how coverage changed

FileStatementsMissingCoverageCoverage
(new stmts)		Lines missing
  sharktank/sharktank/examples
  export_paged_llm_v1.py 81, 97, 113-115
  sharktank/sharktank/models/llm
  config.py
  sharktank/sharktank/utils
  llm_utils.py
Project Total  

This report was generated by python-coverage-comment-action

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

Reviewers

@zeeshanhaque21 zeeshanhaque21 Awaiting requested review from zeeshanhaque21

@rsuderman rsuderman Awaiting requested review from rsuderman

At least 1 approving review is required to merge this pull request.

Assignees

No one assigned

Labels

None yet

Projects

None yet

Milestone

No milestone

Development

Successfully merging this pull request may close these issues.

2 participants

@archana-ramalingam