diff --git a/backends/qualcomm/_passes/annotate_and_quant_scalar.py b/backends/qualcomm/_passes/annotate_and_quant_scalar.py index 86475c39b12..9daaa4aa624 100644 --- a/backends/qualcomm/_passes/annotate_and_quant_scalar.py +++ b/backends/qualcomm/_passes/annotate_and_quant_scalar.py @@ -36,6 +36,7 @@ class AnnotateAndQuantScalar(ExportPass): torch.ops.aten.sub.Scalar, torch.ops.aten.mul.Scalar, torch.ops.aten.div.Scalar, + torch.ops.aten.mul.Tensor, "add", "sub", "mul", diff --git a/backends/qualcomm/_passes/decompose_silu.py b/backends/qualcomm/_passes/decompose_silu.py index d40f13a5923..890d965dec2 100644 --- a/backends/qualcomm/_passes/decompose_silu.py +++ b/backends/qualcomm/_passes/decompose_silu.py @@ -22,7 +22,9 @@ def _copy_meta(self, meta: Dict): def call(self, graph_module: torch.fx.GraphModule): graph = graph_module.graph - partitions = get_source_partitions(graph, [torch.nn.functional.silu]) + partitions = get_source_partitions( + graph, [torch.nn.functional.silu, torch.ops.aten.silu.default] + ) for _, src_partitions in partitions.items(): for src_partition in src_partitions: