onnx-ir 0.0.1__tar.gz

onnx_ir-0.0.1/LICENSE

@@ -0,0 +1,22 @@
+MIT License
+
+Copyright (c) 2025 Justin Chu
+Copyright (c) Microsoft Corporation
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

onnx_ir-0.0.1/MANIFEST.in

@@ -0,0 +1 @@
+global-exclude *_test.py

onnx_ir-0.0.1/PKG-INFO

@@ -0,0 +1,73 @@
+Metadata-Version: 2.2
+Name: onnx-ir
+Version: 0.0.1
+Author-email: Justin Chu <justinchu@microsoft.com>
+License: MIT License
+        
+        Copyright (c) 2025 Justin Chu
+        Copyright (c) Microsoft Corporation
+        
+        Permission is hereby granted, free of charge, to any person obtaining a copy
+        of this software and associated documentation files (the "Software"), to deal
+        in the Software without restriction, including without limitation the rights
+        to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+        copies of the Software, and to permit persons to whom the Software is
+        furnished to do so, subject to the following conditions:
+        
+        The above copyright notice and this permission notice shall be included in all
+        copies or substantial portions of the Software.
+        
+        THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+        IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+        FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+        AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+        LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+        OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+        SOFTWARE.
+        
+Classifier: Development Status :: 4 - Beta
+Classifier: Environment :: Console
+Classifier: Intended Audience :: Developers
+Classifier: Operating System :: POSIX
+Classifier: Operating System :: MacOS :: MacOS X
+Classifier: Operating System :: Microsoft :: Windows
+Classifier: Programming Language :: Python :: 3.8
+Classifier: Programming Language :: Python :: 3.9
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Programming Language :: Python :: 3.13
+Classifier: License :: OSI Approved :: MIT License
+Requires-Python: >=3.8
+Description-Content-Type: text/markdown
+License-File: LICENSE
+Requires-Dist: numpy
+Requires-Dist: onnx>=1.16
+Requires-Dist: typing_extensions>=4.10
+Requires-Dist: ml_dtypes
+
+# ONNX IR
+
+> [!NOTE]
+> This is an adaptation of the `onnxscript.ir` module from the ONNX Script project: https://github.com/microsoft/onnxscript/tree/main/onnxscript/ir
+
+An in-memory IR that supports the full ONNX spec, designed for graph construction, analysis and transformation.
+
+## Features ✨
+
+- Full ONNX spec support: all valid models representable by ONNX protobuf, and a subset of invalid models (so you can load and fix them).
+- Low memory footprint: mmap'ed external tensors; unified interface for ONNX TensorProto, Numpy arrays and PyTorch Tensors etc. No tensor size limitation. Zero copies.
+- Straightforward access patterns: Access value information and traverse the graph topology at ease.
+- Robust mutation: Create as many iterators as you like on the graph while mutating it.
+- Speed: Performant graph manipulation, serialization/deserialization to Protobuf.
+- Pythonic and familiar APIs: Classes define Pythonic apis and still map to ONNX protobuf concepts in an intuitive way.
+- No protobuf dependency: The IR does not require protobuf once the model is converted to the IR representation, decoupling from the serialization format.
+
+## Code Organization 🗺️
+
+- [`_protocols.py`](_protocols.py): Interfaces defined for all entities in the IR.
+- [`_core.py`](_core.py): Implementation of the core entities in the IR, including `Model`, `Graph`, `Node`, `Value`, and others.
+- [`_enums.py`](_enums.py): Definition of the type enums that correspond to the `DataType` and `AttributeType` in `onnx.proto`.
+- [`_name_authority.py`](_name_authority.py): The authority for giving names to entities in the graph, used internally.
+- [`_linked_list.py`](_linked_list.py): The data structure as the node container in the graph that supports robust iteration and mutation. Internal.
+- [`_metadata.py`](_metadata.py): Metadata store for all entities in the IR.

onnx_ir-0.0.1/README.md

@@ -0,0 +1,25 @@
+# ONNX IR
+
+> [!NOTE]
+> This is an adaptation of the `onnxscript.ir` module from the ONNX Script project: https://github.com/microsoft/onnxscript/tree/main/onnxscript/ir
+
+An in-memory IR that supports the full ONNX spec, designed for graph construction, analysis and transformation.
+
+## Features ✨
+
+- Full ONNX spec support: all valid models representable by ONNX protobuf, and a subset of invalid models (so you can load and fix them).
+- Low memory footprint: mmap'ed external tensors; unified interface for ONNX TensorProto, Numpy arrays and PyTorch Tensors etc. No tensor size limitation. Zero copies.
+- Straightforward access patterns: Access value information and traverse the graph topology at ease.
+- Robust mutation: Create as many iterators as you like on the graph while mutating it.
+- Speed: Performant graph manipulation, serialization/deserialization to Protobuf.
+- Pythonic and familiar APIs: Classes define Pythonic apis and still map to ONNX protobuf concepts in an intuitive way.
+- No protobuf dependency: The IR does not require protobuf once the model is converted to the IR representation, decoupling from the serialization format.
+
+## Code Organization 🗺️
+
+- [`_protocols.py`](_protocols.py): Interfaces defined for all entities in the IR.
+- [`_core.py`](_core.py): Implementation of the core entities in the IR, including `Model`, `Graph`, `Node`, `Value`, and others.
+- [`_enums.py`](_enums.py): Definition of the type enums that correspond to the `DataType` and `AttributeType` in `onnx.proto`.
+- [`_name_authority.py`](_name_authority.py): The authority for giving names to entities in the graph, used internally.
+- [`_linked_list.py`](_linked_list.py): The data structure as the node container in the graph that supports robust iteration and mutation. Internal.
+- [`_metadata.py`](_metadata.py): Metadata store for all entities in the IR.

onnx_ir-0.0.1/pyproject.toml

@@ -0,0 +1,106 @@
+[build-system]
+requires = ["setuptools"]
+build-backend = "setuptools.build_meta"
+
+[project]
+name = "onnx-ir"
+version = "0.0.1"
+authors = [{ name = "Justin Chu", email = "justinchu@microsoft.com" }]
+readme = "README.md"
+requires-python = ">=3.8"
+license = { file = "LICENSE" }
+classifiers = [
+  "Development Status :: 4 - Beta",
+  "Environment :: Console",
+  "Intended Audience :: Developers",
+  "Operating System :: POSIX",
+  "Operating System :: MacOS :: MacOS X",
+  "Operating System :: Microsoft :: Windows",
+  "Programming Language :: Python :: 3.8",
+  "Programming Language :: Python :: 3.9",
+  "Programming Language :: Python :: 3.10",
+  "Programming Language :: Python :: 3.11",
+  "Programming Language :: Python :: 3.12",
+  "Programming Language :: Python :: 3.13",
+  "License :: OSI Approved :: MIT License",
+]
+dependencies = ["numpy", "onnx>=1.16", "typing_extensions>=4.10", "ml_dtypes"]
+
+[tool.ruff]
+line-length = 95
+target-version = "py38"
+
+[tool.ruff.lint]
+select = [
+    "B", # flake8-bugbear
+    "C4", # flake8-comprehensions
+    "D", # pydocstyle
+    "E", # pycodestyle
+    "F", # Pyflakes
+    "G", # flake8-logging-format
+    "I", # isort
+    "ISC", # flake8-implicit-str-concat
+    "LOG", # flake8-logging
+    "N", # pep8-naming
+    "NPY", # modern numpy
+    "PERF", # Perflint
+    "PIE", # flake8-pie
+    "PYI", # flake8-pyi
+    "RUF", # Ruff-specific rules
+    "SIM", # flake8-simplify
+    "SLOT", # flake8-slot
+    "T10", # flake8-debugger
+    "TID", # Disallow relative imports
+    "TRY", # flake8-try-except-raise
+    "UP", # pyupgrade
+    "W", # pycodestyle
+    "YTT", # flake8-2020
+]
+# Select preview rules
+preview = true
+ignore = [
+    "B9",  # Opinionated bugbear rules
+    "C408", # Sometimes it is preferable when we construct kwargs
+    "D1", # D1 is for missing docstrings, which is not yet enforced.
+    "D202", # D202 Too strict. "No blank lines allowed after function docstring"
+    "D205", # D205 Too strict. "1 blank line required between summary line and description"
+    "D212",
+    "D400",
+    "D401", # First line of docstring should be in imperative mood
+    "D415", # D415 Not yet enforced. "First line should end with a period, question mark, or exclamation point"
+    "E1", "E2", "E3", # Pycodestyle formatting rules that conflicts with the formatter
+    "E501", # Line length. Not enforced because black will handle formatting
+    "SIM103", # "Return the condition directly" obscures logic sometimes
+    "N802", # Nxx: ONNX Script function sometimes use upper case for names.
+    "N803",
+    "N806",
+    "N999", # Invalid module name
+    "NPY002", # We may not always need a generator
+    "PERF203", # try-except in loops sometimes necessary
+    "PERF401", # List comprehension is not always readable
+    "PYI041", # int | float is more clear
+    "RUF022", # We don't need to sort __all__ for elements to be grouped
+    "RUF031", # Parentheses for tuple in subscripts is more readable
+    "RUF052", # Variables with `_` prefix may not be dummy variables in all cases
+    "SIM102", # Collapible if statements are not always more readable
+    "SIM108", # We don't always encourage ternary operators
+    "SIM114", # Don't always combine if branches for debugability
+    "SIM116", # Don't use dict lookup to replace if-else
+    "TRY003", # Messages can be constructed in the exception
+]
+ignore-init-module-imports = true
+
+[tool.ruff.lint.flake8-tidy-imports.banned-api]
+"pathlib".msg = "Using pathlib can impact performance. Use os.path instead"
+
+[tool.ruff.lint.per-file-ignores]
+"__init__.py" = ["TID252"] # Allow relative imports in init files
+"setup.py" = ["TID251"] # pathlib is allowed in supporting code
+"**/*_test.py" = ["TID251"] # pathlib is allowed in tests
+
+[tool.ruff.lint.flake8-tidy-imports]
+# Disallow all relative imports.
+ban-relative-imports = "all"
+
+[tool.ruff.lint.pydocstyle]
+convention = "google"

onnx_ir-0.0.1/setup.cfg

@@ -0,0 +1,4 @@
+[egg_info]
+tag_build = 
+tag_date = 0
+

onnx_ir-0.0.1/src/onnx_ir/__init__.py

@@ -0,0 +1,154 @@
+# Copyright (c) Microsoft Corporation.
+# Licensed under the MIT License.
+"""In-memory intermediate representation for ONNX graphs."""
+
+__all__ = [
+    # Modules
+    "serde",
+    # IR classes
+    "Tensor",
+    "ExternalTensor",
+    "StringTensor",
+    "SymbolicDim",
+    "Shape",
+    "TensorType",
+    "OptionalType",
+    "SequenceType",
+    "SparseTensorType",
+    "TypeAndShape",
+    "Value",
+    "Attr",
+    "RefAttr",
+    "Node",
+    "Function",
+    "Graph",
+    "GraphView",
+    "Model",
+    # Constructors
+    "AttrFloat32",
+    "AttrFloat32s",
+    "AttrGraph",
+    "AttrGraphs",
+    "AttrInt64",
+    "AttrInt64s",
+    "AttrSparseTensor",
+    "AttrSparseTensors",
+    "AttrString",
+    "AttrStrings",
+    "AttrTensor",
+    "AttrTensors",
+    "AttrTypeProto",
+    "AttrTypeProtos",
+    "Input",
+    # Protocols
+    "ArrayCompatible",
+    "DLPackCompatible",
+    "TensorProtocol",
+    "ValueProtocol",
+    "ModelProtocol",
+    "NodeProtocol",
+    "GraphProtocol",
+    "GraphViewProtocol",
+    "AttributeProtocol",
+    "ReferenceAttributeProtocol",
+    "SparseTensorProtocol",
+    "SymbolicDimProtocol",
+    "ShapeProtocol",
+    "TypeProtocol",
+    "MapTypeProtocol",
+    "FunctionProtocol",
+    # Enums
+    "AttributeType",
+    "DataType",
+    # Types
+    "OperatorIdentifier",
+    # Protobuf compatible types
+    "TensorProtoTensor",
+    # Conversion functions
+    "from_proto",
+    "to_proto",
+    # IR Tensor initializer
+    "tensor",
+    # Pass infrastructure
+    "passes",
+    "traversal",
+    # IO
+    "load",
+    "save",
+]
+
+from onnx_ir import passes, serde, traversal
+from onnx_ir._convenience import tensor
+from onnx_ir._core import (
+    Attr,
+    AttrFloat32,
+    AttrFloat32s,
+    AttrGraph,
+    AttrGraphs,
+    AttrInt64,
+    AttrInt64s,
+    AttrSparseTensor,
+    AttrSparseTensors,
+    AttrString,
+    AttrStrings,
+    AttrTensor,
+    AttrTensors,
+    AttrTypeProto,
+    AttrTypeProtos,
+    ExternalTensor,
+    Function,
+    Graph,
+    GraphView,
+    Input,
+    Model,
+    Node,
+    OptionalType,
+    RefAttr,
+    SequenceType,
+    Shape,
+    SparseTensorType,
+    StringTensor,
+    SymbolicDim,
+    Tensor,
+    TensorType,
+    TypeAndShape,
+    Value,
+)
+from onnx_ir._enums import (
+    AttributeType,
+    DataType,
+)
+from onnx_ir._io import load, save
+from onnx_ir._protocols import (
+    ArrayCompatible,
+    AttributeProtocol,
+    DLPackCompatible,
+    FunctionProtocol,
+    GraphProtocol,
+    GraphViewProtocol,
+    MapTypeProtocol,
+    ModelProtocol,
+    NodeProtocol,
+    OperatorIdentifier,
+    ReferenceAttributeProtocol,
+    ShapeProtocol,
+    SparseTensorProtocol,
+    SymbolicDimProtocol,
+    TensorProtocol,
+    TypeProtocol,
+    ValueProtocol,
+)
+from onnx_ir.serde import TensorProtoTensor, from_proto, to_proto
+
+
+DEBUG: bool = False
+
+
+def __set_module() -> None:
+    """Set the module of all functions in this module to this public module."""
+    global_dict = globals()
+    for name in __all__:
+        global_dict[name].__module__ = __name__
+
+
+__set_module()