Add support for Conv-Act-Pool fusion

Signed-off-by: gcunhase <[email protected]>

Author: gcunhase

modelopt/onnx/quantization/partitioning.py

@@ -45,8 +45,9 @@ def _build_fusible_partition(
 
     Add a nodes to the partition if any of these holds:
     1. The node is a unary or binary pointwise operation and fusible by cask
-    2. The node is BN and/or Relu and fusible with preceding Conv op
-    3. The node is a residual Add and fusible with current partition
+    2. The node is BN and/or Relu and fusible with preceding Conv op (Conv-Act fusion)
+    3. The node is MaxPool following a Conv-Act pattern (Conv-Act-Pool fusion)
+    4. The node is a residual Add and fusible with current partition
 
     Args:
         cur_node: Current candidate node for the partition.
@@ -135,7 +136,7 @@ def _is_fusible_mul(mul_node: Node) -> bool:
                 and _is_cask_fusible(consumer_node, partition_node_outputs)
             )
             or (
-                consumer_node.op in ["BatchNormalization", "Relu"]
+                consumer_node.op in ["MaxPool", "BatchNormalization", "Relu"]
                 and get_fusible_backbone(consumer_node, graph)
             )
             or _is_on_non_residual_path(consumer_node)