Skip to content

Allocate blocks from id=1 for HPU#160

Merged
kzawora-intel merged 1 commit intoHabanaAI:habana_mainfrom
kdamaszk:allocate-blocks-from-id-1
Aug 6, 2024
Merged

Allocate blocks from id=1 for HPU#160
kzawora-intel merged 1 commit intoHabanaAI:habana_mainfrom
kdamaszk:allocate-blocks-from-id-1

Conversation

@kdamaszk
Copy link
Copy Markdown

@kdamaszk kdamaszk commented Aug 6, 2024

This PR is related to the change from habana_next: #155

When using BlockManagerV2, blocks are allocated from id=0, which can cause undefined behavior in reuse_and_cache function. This issue is better visible with --tensor-parallel-size > 1.
This change moves the first block id from 0 to 1 for HPU device.

@kdamaszk kdamaszk requested a review from kzawora-intel August 6, 2024 08:43
Copy link
Copy Markdown

@kzawora-intel kzawora-intel left a comment

Choose a reason for hiding this comment

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

lgtm!

@kzawora-intel kzawora-intel merged commit b43c7f9 into HabanaAI:habana_main Aug 6, 2024
kzawora-intel added a commit that referenced this pull request Aug 6, 2024
kzawora-intel added a commit that referenced this pull request Aug 6, 2024
nirda7 pushed a commit that referenced this pull request Aug 6, 2024
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