Skip to content

model: decoderonly batch_attn_decode opt - #559

Open
rebel-thkim wants to merge 7 commits into
devfrom
batch_attn_decode
Open

model: decoderonly batch_attn_decode opt#559
rebel-thkim wants to merge 7 commits into
devfrom
batch_attn_decode

Conversation

@rebel-thkim

Copy link
Copy Markdown
Contributor

Pull Request Description

⚠️ Important: Branch Target

  • New features, enhancements, and non-critical fixes: Merge to dev branch
  • Critical hotfixes only: Merge to main branch (must also merge to dev)

Please ensure you've selected the correct base branch before submitting!

Type of Change

  • Release (dev → main merge for production release)
  • New Model Support
  • Bug fix (non-breaking change which fixes an issue)
  • New feature (non-breaking change which adds functionality)
  • Breaking change (fix or feature that would cause existing functionality to not work as expected)
  • Documentation update
  • Performance improvement
  • Code refactoring
  • Other (please describe):

Changes Overview

Motivation and Context

Related Issues


Conventional commit

type(optional scope): description

Type candidate

  • Model Updates
    • model: Adding New models or Bugfix for existing models
      • ex) Add LlavaNext
      • ex) Bugfix Whisper
  • Enhancements
    • performance: Optimizing some models or this library itself
      • ex) Loading RBLNModel faster
      • ex) Optimizing Memory Usage of DecoderOnlyModel
  • Code Refactor
    • refactor: Re-arrange class architecture, or more.
      • ex) Refactor Seq2Seq
  • Documentation
    • doc: Update docstring only
  • Library Dependencies
    • dependency: Update requirements, something like that.
  • Release
    • release: Merging dev to main for production release
      • ex) Release v1.2.0
  • Other
    • other: None of above.
      • ex) ci update
      • ex) pdm update

Comment on lines +36 to +38
def _is_batch_attn_opt_enabled() -> bool:
"""Check whether VLLM_RBLN_BATCH_ATTN_OPT env var is set to enable batched dynamic decode."""
return os.environ.get("VLLM_RBLN_BATCH_ATTN_OPT", "0") == "1"

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.

(note) @rebel-jswoo

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.

3 participants