Rebase main

sahas3 · sahas3 · commit 8a982db8b17c · 2024-12-23T09:18:58.000-05:00
diff --git a/python/torch_mlir/extras/fx_decomp_util.py b/python/torch_mlir/extras/fx_decomp_util.py
@@ -49,8 +49,6 @@
     torch.ops.aten.nan_to_num.default,
     torch.ops.aten.unbind,
     torch.ops.aten.diag,
-    torch.ops.aten.lstm.input,
-    torch.ops.aten.gru.input,
 ]
 if hasattr(torch.ops.aten, "_scaled_dot_product_flash_attention_for_cpu"):
     DEFAULT_DECOMPOSITIONS.append(

Original file line number	Diff line number	Diff line change
`@@ -49,8 +49,6 @@`
`49`	`49`	`torch.ops.aten.nan_to_num.default,`
`50`	`50`	`torch.ops.aten.unbind,`
`51`	`51`	`torch.ops.aten.diag,`
`52`		`- torch.ops.aten.lstm.input,`
`53`		`- torch.ops.aten.gru.input,`
`54`	`52`	`]`
`55`	`53`	`if hasattr(torch.ops.aten, "_scaled_dot_product_flash_attention_for_cpu"):`
`56`	`54`	`DEFAULT_DECOMPOSITIONS.append(`