Skip to content

perf(llama32_1b): dual-hart K-split - #221

Open
karabambus wants to merge 1 commit into
aifoundry-org:mainfrom
karabambus:llama32-submission
Open

perf(llama32_1b): dual-hart K-split#221
karabambus wants to merge 1 commit into
aifoundry-org:mainfrom
karabambus:llama32-submission

Conversation

@karabambus

Copy link
Copy Markdown
Contributor

Changes

The N=1 decode dot product for Q8_0 previously gave each hart its own independent output rows. This change splits along K instead: the two sibling harts of a minion share a row and each computes half of the reduction. One partial is published through the shire-local L2 scratchpad and handed over with an FCC semaphore handshake; hart 0 sums the two halves and stores the result.

Results

Board measurements, llama-server decode, Llama-3.2-1B-Instruct Q8_0:

variant tok/s PPL
baseline 12.9514 15.3098
this change 13.8937 15.3098

Recipe

Unfortunately I didn't have opportunity to properly work on this, I studied #159 and made some improvements and implemented it on llama and that is basically it.

@karabambus
karabambus requested a review from AFOliveira as a code owner July 26, 2026 07:40
@github-actions github-actions Bot added track: week-2-challenge Week 2 focused hardware challenge track: llama-3.2-1b-performance Global Llama 3.2 1B performance prize labels Jul 26, 2026
@karabambus
karabambus marked this pull request as draft July 26, 2026 08:25
@karabambus
karabambus marked this pull request as ready for review July 26, 2026 08:25
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

track: llama-3.2-1b-performance Global Llama 3.2 1B performance prize track: week-2-challenge Week 2 focused hardware challenge

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant