Skip to content

webgpu: Fuse activations into the im2col-matmul Conv path - #32334

Closed
xhcao wants to merge 1 commit into
microsoft:mainfrom
xhcao:img2col_fused
Closed

xhcao wants to merge 1 commit into
microsoft:mainfrom
xhcao:img2col_fused

Conversation

@xhcao

@xhcao xhcao commented Aug 31, 2026

Copy link
Copy Markdown
Contributor

Description

The im2col-matmul Conv kernel used to bail out when the Conv node had a fused
activation. This PR applies the activation in the im2col-matmul shader, right
after the bias is added, so fused Conv can use this path too.

Motivation and Context

Fused Conv is common in vision models, and excluding it meant those models never
got the faster kernel. Measured on PTL, this gives ~15% improvement on
detr-resnet-50-fp16.onnx and resnet50-v1-f16-demo.onnx.

@azure-pipelines

Copy link
Copy Markdown
Azure Pipelines:
There may be pipelines that require an authorized user to comment /azp run to run.

@xhcao

xhcao commented Aug 31, 2026

Copy link
Copy Markdown
Contributor Author

Jianhui Dai (@daijh) Jie Chen (@jchen10), Jianhui, PTAL, thanks.

@qjia7

Copy link
Copy Markdown
Contributor

Is this one duplicated with #32185?

@xhcao

xhcao commented Sep 1, 2026

Copy link
Copy Markdown
Contributor Author

Is this one duplicated with #32185?
Yes, I will close it.

@xhcao xhcao closed this Sep 1, 2026
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