diff --git a/backends/qualcomm/utils/utils.py b/backends/qualcomm/utils/utils.py index 1bcfa3a6f6b..f94e22c5306 100644 --- a/backends/qualcomm/utils/utils.py +++ b/backends/qualcomm/utils/utils.py @@ -287,6 +287,10 @@ def set_spec(module, options): return max_sf_size +def canonicalize_program(obj): + update_spill_fill_size(obj) + + def get_decomp_table() -> Dict[torch._ops.OperatorBase, Callable]: source_decompositions = core_aten_decompositions() # The below super ops are supported by QNN