Update decomp table (#8210)

pytorch · Oct 3, 2024 · a2681e8 · a2681e8
1 parent c22dfe9
commit a2681e8
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/torch_xla/stablehlo.py b/torch_xla/stablehlo.py
@@ -320,8 +320,8 @@ def _extract_input_args(exported_model, options):
 
 
 def _run_decompositions(exported_model):
-  if hasattr(torch.export, 'core_op_decompositions'):
-    decomp_table = torch.export.core_op_decompositions()
+  if hasattr(torch.export, 'default_decompositions'):
+    decomp_table = torch.export.default_decompositions()
   else:
     decomp_table = core_aten_decompositions()
   decomp_table.update(_extra_decompositions)