The issue does not break the current 0429 nightly dependencies. This is a fix for forward compatibility and future dependencies upgrade.
Starting from torch 2.5.0 nightly (after mid June 2024), torch.nn.functional.interpolate no longer gets decomposed right after export but exported to torch.ops.aten.upsample_nearest2d.vec and torch.ops.aten.upsample_bilinear2d.vec in the first step. This behavior breaks all our pattern matching composite builder.
The fix in this PR runs decompositions on patterns and exported model for these ops. A more stable future work would be relying on aten composite builder, instead of decomposing then pattern matching.
The issue does not break the current 0429 nightly dependencies. This is a fix for forward compatibility and future dependencies upgrade.
Starting from torch 2.5.0 nightly (after mid June 2024),
torch.nn.functional.interpolateno longer gets decomposed right after export but exported totorch.ops.aten.upsample_nearest2d.vecandtorch.ops.aten.upsample_bilinear2d.vecin the first step. This behavior breaks all our pattern matching composite builder.The fix in this PR runs decompositions on patterns and exported model for these ops. A more stable future work would be relying on aten composite builder, instead of decomposing then pattern matching.
BUG=https://b.corp.google.com/issues/348413649