Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Unbreak test models llama CI (#6026)
Summary: Pull Request resolved: #6026 Did a bunch of debugging on OSS CI:https://github.com/pytorch/executorch/actions/runs/11241297226/job/31252590975 Was able to confirm although the problem happens in `ConvertToLinear` but the root cause is we are partitioning the graph differently between these two pytorch nightly: dev20240916 and dev20240917. The exported graph looks the same but the partitioner was behaving differently and causes the `ConvertToLinear` pass to error out. We can't really revert back to dev20240916 nightly because it breaks other CI jobs, see #5987. The current approach I'm taking avoids decomposing linear by using `to_edge_lower_and_transform` API. This avoids jumping into the rabbit hole of debugging the partitioning & tagging logic. Reviewed By: digantdesai, Jack-Khuu, tugsbayasgalan Differential Revision: D64074891 fbshipit-source-id: c434f9f5fc240b9268a7419fc66ee4a365ae1664
- Loading branch information