add a list for ops to be added

Differential Revision: D61002938

Pull Request resolved: #4624

Author: cccclai

backends/qualcomm/partition/common_defs.py

@@ -16,6 +16,10 @@
     exir_ops.edge.aten.copy.default,
 ]
 
+to_be_implemented_operator = [
+    exir_ops.edge.aten.where.default,
+]
+
 allow_list_operator = [
     _operator.getitem,
 ]

backends/qualcomm/partition/qnn_partitioner.py

@@ -27,7 +27,11 @@
 from torch.fx.passes.infra.partitioner import Partition
 from torch.fx.passes.operator_support import OperatorSupportBase
 
-from .common_defs import allow_list_operator, not_supported_operator
+from .common_defs import (
+    allow_list_operator,
+    not_supported_operator,
+    to_be_implemented_operator,
+)
 
 
 class QnnOperatorSupport(OperatorSupportBase):
@@ -62,6 +66,12 @@ def is_node_supported(self, _, node: torch.fx.Node) -> bool:
         if node.op != "call_function" or node.target in not_supported_operator:
             return False
 
+        if node.target in to_be_implemented_operator:
+            print(
+                f"[QNN Partitioner Op Support]: {node.target.__name__} | Skipped, this op can be supported, please report an issue in https://github.com/pytorch/executorch/issues"
+            )
+            return False
+
         if node.target in allow_list_operator:
             return True