Skip to content

spec : add DFlash2 support (local convolution + candidate selector) (#27342) - #27816

Merged
ngxson merged 2 commits into
masterfrom
xsn/dflash2
Aug 27, 2026
Merged

spec : add DFlash2 support (local convolution + candidate selector) (#27342)#27816
ngxson merged 2 commits into
masterfrom
xsn/dflash2

Conversation

@ngxson

@ngxson ngxson commented Aug 27, 2026

Copy link
Copy Markdown
Collaborator

I cannot push to #27342 because the PR was created from an org, applying @ORippler 's request to revert top-k.cu here

for full changes, please refer to #27342

SubSir and others added 2 commits August 27, 2026 19:05
…27342)

* support DFlash2

* Add p_min in DFlash2

Assisted-by: Claude Opus 5

* Revert unnecessary changes

Assisted-by: Claude Opus 5

* Revert draft sampling in rejection sampling

Assisted-by: Claude Opus 5

* Refactor code structure

Assisted-by: Claude Opus 5

* Delete embedding scaling

Assisted-by: Claude Opus 5

* Gate output transforms on DFlash2

Assisted-by: Claude Opus 5

* Optimize Dflash 2 cost

Assisted-by: Claude Opus 5

* Avoid using atoi

Assisted-by: Claude Opus 5

* Modify comments

Assisted-by: Claude Opus 5

* Move llama_model_dflash_selector_top_k to llama-ext.h

Assisted-by: Claude Opus 5

* Formatting

Assisted-by: Claude Opus 5

* Apply patch to fix the mrope bug

Assisted-by: Claude Opus 5

* fix ci

Assisted-by: Claude Opus 5

* Fix graph number calculation

Assisted-by: Claude Opus 5

* rename hid and unary

Assisted-by: Claude Opus 5

---------

Co-authored-by: Jian Chen <jianchen0311@gmail.com>
Co-authored-by: Xuan-Son Nguyen <son@huggingface.co>
@github-actions github-actions Bot added model Model specific testing Everything test related conversion labels Aug 27, 2026
@ngxson
ngxson marked this pull request as ready for review August 27, 2026 17:16
@ngxson
ngxson requested review from a team, CISC and ggerganov as code owners August 27, 2026 17:16
@ngxson
ngxson merged commit b10f9ca into master Aug 27, 2026
25 of 28 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

conversion model Model specific testing Everything test related

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants