onnxruntime
WebGPU JSEP: Make shader code not depend on input broadcasting patterns
#22536

Merged

WebGPU JSEP: Make shader code not depend on input broadcasting patterns #22536

guschmue merged 8 commits into microsoft:main from jiangzhaoming:MatMulShaderCacheKeyBroadcasting0

WebGPU JSEP: Add inputs broadcasting into MatMul shader cache key

1ee2d5e4

qjia7 commented on 2024-10-22

Change the shader implementation to use only ranks and uniform

6d225e01

jiangzhaoming changed the title ~~WebGPU JSEP: Add inputs broadcasting into MatMul shader cache key~~ WebGPU JSEP: Make shader code not rely on input broadcasting patterns 1 year ago

jiangzhaoming changed the title ~~WebGPU JSEP: Make shader code not rely on input broadcasting patterns~~ WebGPU JSEP: Make shader code not depend on input broadcasting patterns 1 year ago

qjia7 commented on 2024-10-24

Improve code reusing

74c4d225

Make helper only handle the batch indices

b70e48c3

guschmue added ep:WebGPU

qjia7 commented on 2024-10-28

Separate shaders impl from matmul.ts (OP impl) to avoid circular depe…

d30b5b54

qjia7 commented on 2024-11-01

Remove unused getBroadcastDims

54ac6227

jiangzhaoming requested a review from

fs-eire 1 year ago

guschmue dismissed these changes on 2024-11-08

Merge branch 'main' into MatMulShaderCacheKeyBroadcasting0

bbaa003b

Fix merging issue

84fd707e

jiangzhaoming dismissed their stale review via 84fd707e 1 year ago

guschmue approved these changes on 2024-11-08

guschmue merged d9b91682 into main 1 year ago

Reviewers

guschmue

qjia7

fs-eire

Assignees

No one assigned

Labels

ep:WebGPU

Milestone

No milestone

onnxruntime WebGPU JSEP: Make shader code not depend on input broadcasting patterns #22536 Merged

WebGPU JSEP: Make shader code not depend on input broadcasting patterns #22536

onnxruntime
WebGPU JSEP: Make shader code not depend on input broadcasting patterns
#22536

Merged