8000 [async TP] insert reshape node to handle "reshape -> scaled mm -> reshape pattern" in async TP with rowwise scales by danielvegamyhre · Pull Request #148001 · pytorch/pytorch · GitHub
[go: up one dir, main page]
Skip to content

[async TP] insert reshape node to handle "reshape -> scaled mm -> reshape pattern" in async TP with rowwise scales#148001

Closed
danielvegamyhre wants to merge 1 commit intopytorch:mainfrom
danielvegamyhre:swap

Commits

Commits on Mar 1, 2025
0