[ET-VK][int4] Wrap int4 linear calls with view_copy nodes to squeeze/unsqueeze inputs
Pull Request resolved: https://github.com/pytorch/executorch/pull/8226 This is done automatically for full-precision linear/mm nodes in the graph at torch.export graph tracing time, but is not done for the int4 op. The new pass adds view_copy nodes, as there are subsequent passes which can fuse view_copy nodes if redundant, and convert view_copy nodes to squeeze/unsqueeze nodes. ghstack-source-id: 264952606 @exported-using-ghexport Differential Revision: [D69065866](https://our.internmc.facebook.com/intern/diff/D69065866/) --------- Co-authored-by: Nathanael See <nathanaelsee@fb.com>
P
pytorchbot committed
e79713ebe9f2984e922883e7c2a9817103a187ed
Parent: 8f0d797
Committed by GitHub <noreply@github.com>
on 2/6/2025, 6:16:35 PM