[Pass] Graph extractor pass

[justinchuby]

• Added a new pass for extracting subgraphs from a graph (ExtractGraphPass) in onnxscript/ir/passes/common/graph_extration.py. This pass allows for the extraction of a sub-model defined by specific input and output tensor names.

[Copilot]

Pull Request Overview

This PR introduces a new pass for extracting subgraphs from an ONNX model based on exact input and output tensor names. Key changes include:

• Adding a new pass implementation in onnxscript/ir/passes/common/graph_extration.py.
• Enhancing pre- and post-condition checks in the pass infrastructure in onnxscript/ir/passes/_pass_infra.py.
• Exporting the new create_value_mapping function via the convenience modules.

Reviewed Changes

Copilot reviewed 4 out of 4 changed files in this pull request and generated 1 comment.

File	Description
onnxscript/ir/passes/common/graph_extration.py	Implements the new ExtractGraphPass for subgraph extraction.
onnxscript/ir/passes/_pass_infra.py	Updates PassBase and PassManager for improved invariant checking.
onnxscript/ir/_convenience.py	Exports create_value_mapping and replace_nodes_and_values.
onnxscript/ir/convenience.py	Imports and re-exports create_value_mapping.

Comments suppressed due to low confidence (1)

onnxscript/ir/passes/common/graph_extration.py:1

• The filename 'graph_extration.py' appears to be misspelled; consider renaming it to 'graph_extraction.py' for clarity.

# Copyright (c) Microsoft Corporation.

[codecov]

❌ 8 Tests Failed:

Tests completed	Failed	Passed	Skipped
7678	8	7670	4113

View the top 3 failed test(s) by shortest run time

onnxscript.ir.passes.common.graph_extration_test.TestExtractGraphPass::test_extract_subgraph

Stack Traces | 0.002s run time

.../passes/common/graph_extration_test.py:34: in test_extract_subgraph
    self.assertEqual(len(result.model.graph.nodes), 2)
E   AttributeError: 'Graph' object has no attribute 'nodes'

onnxscript.ir.passes.common.graph_extration_test.TestExtractGraphPass::test_extract_subgraph_with_initializers

Stack Traces | 0.003s run time

.../passes/common/graph_extration_test.py:65: in test_extract_subgraph_with_initializers
    self.assertEqual(len(result.model.graph.nodes), 3)
E   AttributeError: 'Graph' object has no attribute 'nodes'

onnxscript.ir.passes.common.graph_extration_test.TestExtractGraphPass::test_extract_subgraph_with_subgraph

Stack Traces | 0.004s run time

.../passes/common/graph_extration_test.py:119: in test_extract_subgraph_with_subgraph
    self.assertEqual(len(result.model.graph.nodes), 1)
E   AttributeError: 'Graph' object has no attribute 'nodes'

To view more test analytics, go to the Test Analytics Dashboard
_{📋 Got 3 mins? Take this short survey to help us improve Test Analytics.}

[justinchuby]

Change the copilot output