From d3df097ff074c8ecfb174d9bd633e5518e6e8394 Mon Sep 17 00:00:00 2001 From: Manuel Candales Date: Fri, 13 Mar 2026 16:01:35 -0400 Subject: [PATCH] fix linear bias decomposition invokation --- examples/models/parakeet/export_parakeet_tdt.py | 6 +++--- examples/models/voxtral_realtime/export_voxtral_rt.py | 6 +++--- 2 files changed, 6 insertions(+), 6 deletions(-) diff --git a/examples/models/parakeet/export_parakeet_tdt.py b/examples/models/parakeet/export_parakeet_tdt.py index 6747880cd9e..082f73b5611 100644 --- a/examples/models/parakeet/export_parakeet_tdt.py +++ b/examples/models/parakeet/export_parakeet_tdt.py @@ -513,9 +513,9 @@ def _create_metal_partitioners(programs): # print(f"Running decompositions for {key}") # print(ep.graph_module) if key != "preprocessor": - updated_programs[key] = ep.run_decompositions( - {torch.ops.aten.linear.default: _linear_bias_decomposition} - ) + decomp_table = torch.export.default_decompositions() + decomp_table[torch.ops.aten.linear.default] = _linear_bias_decomposition + updated_programs[key] = ep.run_decompositions(decomp_table) else: updated_programs[key] = ep diff --git a/examples/models/voxtral_realtime/export_voxtral_rt.py b/examples/models/voxtral_realtime/export_voxtral_rt.py index 31f792232a3..edebab67c7b 100644 --- a/examples/models/voxtral_realtime/export_voxtral_rt.py +++ b/examples/models/voxtral_realtime/export_voxtral_rt.py @@ -395,9 +395,9 @@ def lower_to_executorch(programs, metadata, backend="xnnpack"): # Run decompositions for Metal backend updated_programs = {} for key, ep in programs.items(): - updated_programs[key] = ep.run_decompositions( - {torch.ops.aten.linear.default: _linear_bias_decomposition} - ) + decomp_table = torch.export.default_decompositions() + decomp_table[torch.ops.aten.linear.default] = _linear_bias_decomposition + updated_programs[key] = ep.run_decompositions(decomp_table) programs = updated_programs partitioner = {}