onnxruntime
[WebGPU] LinearAttention: increase tile_v when subgroups are available
#28519

Merged

[WebGPU] LinearAttention: increase tile_v when subgroups are available #28519

guschmue merged 6 commits into microsoft:main from daijh:linear-attn-dev

daijh marked this pull request as draft 180 days ago

[WebGPU] LinearAttention: increase tile_v when subgroups are available

5a8762d3

Add comments

e84d6752

guschmue added ep:WebGPU

Gate the tile_v expansion on seq_length >= 16

4cb94d53

daijh force pushed from 58a09c74 to 4cb94d53 176 days ago

daijh marked this pull request as ready for review 176 days ago

qjia7 requested a review from

copilot-pull-request-reviewer 174 days ago

copilot-pull-request-reviewer commented on 2026-05-21

Clamp tile_v to workgroup_size

315460be

qjia7 commented on 2026-05-22

Add TODO to support vec2 (components==2) in linear attention

c8f87882

Simplify components check: remove redundant head_dim_v >= 4 guard

c7520c0c

guschmue approved these changes on 2026-05-22

guschmue merged 1053327e into main 173 days ago

daijh deleted the linear-attn-dev branch 170 days ago

Reviewers

guschmue

qjia7

Assignees

No one assigned

Labels

ep:WebGPU

Milestone

No milestone