emx-onnx-cgen 0.2.0__py3-none-any.whl

emx_onnx_cgen/validation.py

@@ -0,0 +1,76 @@
+from __future__ import annotations
+
+from .errors import ShapeInferenceError
+from .ir.model import Node
+
+
+def normalize_axis(axis: int, shape: tuple[int, ...], node: Node) -> int:
+    if not shape:
+        raise ShapeInferenceError(f"{node.op_type} does not support scalar inputs")
+    rank = len(shape)
+    if axis < 0:
+        axis += rank
+    if axis < 0 or axis >= rank:
+        raise ShapeInferenceError(
+            f"{node.op_type} axis {axis} is out of range for rank {rank}"
+        )
+    return axis
+
+
+def normalize_concat_axis(axis: int, rank: int) -> int:
+    if axis < 0:
+        axis += rank
+    if axis < 0 or axis >= rank:
+        raise ShapeInferenceError(
+            f"Concat axis out of range for rank {rank}: {axis}"
+        )
+    return axis
+
+
+def ensure_output_shape_matches_input(
+    node: Node,
+    input_shape: tuple[int, ...],
+    output_shape: tuple[int, ...],
+) -> None:
+    if input_shape != output_shape:
+        raise ShapeInferenceError(
+            f"{node.op_type} output shape must be {input_shape}, got {output_shape}"
+        )
+
+
+def validate_concat_shapes(
+    input_shapes: tuple[tuple[int, ...], ...],
+    output_shape: tuple[int, ...],
+    axis: int,
+) -> int:
+    ranks = {len(shape) for shape in input_shapes}
+    if len(ranks) != 1:
+        raise ShapeInferenceError(
+            f"Concat inputs must have matching ranks, got {input_shapes}"
+        )
+    rank = ranks.pop()
+    axis = normalize_concat_axis(axis, rank)
+    base_shape = list(input_shapes[0])
+    axis_dim = 0
+    for shape in input_shapes:
+        if len(shape) != rank:
+            raise ShapeInferenceError(
+                f"Concat inputs must have matching ranks, got {input_shapes}"
+            )
+        for dim_index, dim in enumerate(shape):
+            if dim_index == axis:
+                continue
+            if dim != base_shape[dim_index]:
+                raise ShapeInferenceError(
+                    "Concat inputs must match on non-axis dimensions, "
+                    f"got {input_shapes}"
+                )
+        axis_dim += shape[axis]
+    base_shape[axis] = axis_dim
+    expected_output_shape = tuple(base_shape)
+    if output_shape != expected_output_shape:
+        raise ShapeInferenceError(
+            "Concat output shape must be "
+            f"{expected_output_shape}, got {output_shape}"
+        )
+    return axis

emx_onnx_cgen-0.2.0.dist-info/METADATA

@@ -0,0 +1,128 @@
+Metadata-Version: 2.4
+Name: emx-onnx-cgen
+Version: 0.2.0
+Summary: emmtrix ONNX-to-C Code Generator
+Requires-Python: >=3.10
+Description-Content-Type: text/markdown
+
+# emmtrix ONNX-to-C Code Generator (emx-onnx-cgen)
+
+[![PyPI - Version](https://img.shields.io/pypi/v/emx-onnx-cgen.svg)](https://pypi.org/project/emx-onnx-cgen)
+
+`emx-onnx-cgen` compiles ONNX models to portable, deterministic C code for deeply embedded systems. The generated code is designed to run without dynamic memory allocation, operating systems, or external runtimes, making it suitable for safety-critical and resource-constrained targets.
+
+Key characteristics:
+
+- **No dynamic memory allocation** (`malloc`, `free`, heap usage)
+- **Static, compile-time known memory layout** for parameters, activations, and temporaries
+- **Deterministic control flow** (explicit loops, no hidden dispatch or callbacks)
+- **No OS or libc dependencies** beyond basic C
+- **Single-threaded execution model**
+- **Bitwise-stable code generation** for reproducible builds
+- **Readable, auditable C code** suitable for certification and code reviews
+- Designed for **bare-metal and RTOS-based systems**
+
+## Goals
+
+- Correctness-first compilation with outputs comparable to ONNX Runtime.
+- Deterministic and reproducible C code generation.
+- Clean, pass-based compiler architecture (import → normalize → optimize → lower → emit).
+- Minimal C runtime with explicit, predictable data movement.
+
+## Non-goals
+
+- Aggressive performance optimizations in generated C.
+- Implicit runtime dependencies or dynamic loading.
+- Training/backpropagation support.
+
+## Features
+
+- CLI for ONNX-to-C compilation and verification.
+- Deterministic codegen with explicit tensor shapes and loop nests.
+- Minimal C runtime templates in `templates/`.
+- ONNX Runtime comparison for end-to-end validation.
+- Official ONNX operator coverage tracking.
+- Support for a wide range of ONNX operators (see `OFFICIAL_ONNX_FILE_SUPPORT.md`).
+- Supported data types:
+  - `float`, `double`, `float16`
+  - `int8_t`, `uint8_t`, `int16_t`, `uint16_t`, `int32_t`, `uint32_t`, `int64_t`, `uint64_t`
+  - `bool`
+- Supporting dynamic dimensions by utilizing C99 variable-length arrays (VLAs).
+
+## Installation
+
+Install the package directly from PyPI (recommended):
+
+```bash
+pip install emx-onnx-cgen
+```
+
+Optional for verification and tests:
+
+- `onnxruntime`
+- `numpy`
+- A C compiler (`cc`, `gcc`, `clang` or via `--cc`)
+
+## Quickstart
+
+Compile an ONNX model into a C source file:
+
+```bash
+emx-onnx-cgen compile path/to/model.onnx build/model.c
+```
+
+Verify an ONNX model end-to-end against ONNX Runtime:
+
+```bash
+emx-onnx-cgen verify path/to/model.onnx
+```
+
+## CLI Reference
+
+`emx-onnx-cgen` provides two subcommands: `compile` and `verify`.
+
+### `compile`
+
+```bash
+emx-onnx-cgen compile <model.onnx> <output.c> [options]
+```
+
+Options:
+
+- `--template-dir`: Directory containing the C templates (default: `templates`).
+- `--model-name`: Override the generated model name (default: output file stem).
+- `--emit-testbench`: Emit a JSON-producing `main()` testbench for validation.
+- `--emit-data-file`: Emit constant data arrays into a companion `_data` C file.
+- `--no-restrict-arrays`: Disable `restrict` qualifiers on generated array parameters.
+
+### `verify`
+
+```bash
+emx-onnx-cgen verify <model.onnx> [options]
+```
+
+Options:
+
+- `--template-dir`: Directory containing the C templates (default: `templates`).
+- `--model-name`: Override the generated model name (default: model file stem).
+- `--cc`: Explicit C compiler command for building the testbench binary.
+
+## Output
+
+By default, the compiler emits a single C source file that includes:
+
+- A generated entry point that mirrors the ONNX graph inputs/outputs.
+- Tensor buffers for constants and temporaries.
+- A lightweight runtime implemented via templates in `templates/`.
+
+When `--emit-data-file` is enabled, the main C source declares constant arrays
+as `extern`, and a second file named like the output with a `_data` suffix
+contains the constant definitions.
+
+## Official ONNX test coverage
+
+See [`OFFICIAL_ONNX_FILE_SUPPORT.md`](OFFICIAL_ONNX_FILE_SUPPORT.md) for the generated support matrix.
+
+## Maintained by
+
+This project is maintained by [emmtrix](https://www.emmtrix.com).

emx_onnx_cgen-0.2.0.dist-info/RECORD

@@ -0,0 +1,76 @@
+emx_onnx_cgen/__init__.py,sha256=jUSbu1kJ0krzVTYEcph3jCprBhD7tWNtiSdL6r29KrM,221
+emx_onnx_cgen/__main__.py,sha256=iC1lLVtR6-TmpL6OxXcy3oIntExUtajn9-q627R1XyI,140
+emx_onnx_cgen/_build_info.py,sha256=tp1Kzo4PNvj3ydRSCdr84YUd2wEUVUopl7AoaJfOaHk,112
+emx_onnx_cgen/cli.py,sha256=2eQ6JxfdamiLDqo-ZfZNnnb1a1I7NTEvecfUJaAA_3M,11024
+emx_onnx_cgen/compiler.py,sha256=foB9JK1Z2NRcRg_Fn-kL9K7L7tKGWxI9-kaMbYg3dnM,20877
+emx_onnx_cgen/dtypes.py,sha256=jRx3BBvk0qFW14bngoL1B7L_IRasyNJ4jqhpM5YhcOM,1335
+emx_onnx_cgen/errors.py,sha256=HpOv95mTgr9ZX2gYe1RtwVMbPskh7zkqjU_FgAD-uIM,363
+emx_onnx_cgen/onnx_import.py,sha256=aMvSxT3ycg4UmnutWYvsQpzGt2m_KpNnDIiddlg-vDA,9028
+emx_onnx_cgen/ops.py,sha256=pW3ks2EJITiJxvThfU58KjQZE7AFUiPmMwKZRCNb1mY,16586
+emx_onnx_cgen/validation.py,sha256=KFdUdGjQbzTj1szCJcjxnTi8f5l6ywNgCB9abbBpTbM,2360
+emx_onnx_cgen/codegen/__init__.py,sha256=-_sxL87uyAIunaetjUvIUo2bc46ugVlaNtSsidegMRM,362
+emx_onnx_cgen/codegen/c_emitter.py,sha256=gCgbqTgDWGAmp7H8TCFPBKp7DCNb_nskkMCAzKwDG0Y,337438
+emx_onnx_cgen/ir/__init__.py,sha256=fD2D8qxlGoCFJb0m9v6u3XTgzSxDOhB4cfLBiCLovzg,102
+emx_onnx_cgen/ir/model.py,sha256=e8vRA0RNDU8Ioz3TXQKpdUhDtUK6Hm71KouUPMhCcpg,1213
+emx_onnx_cgen/lowering/__init__.py,sha256=wrxLMWcPUH1RbPJOs0Tsdb12FhXjAAeZVDYwKqcIuzw,103
+emx_onnx_cgen/lowering/arg_reduce.py,sha256=2AowDRCJRkIvrVBphbA0rM18oCWEpCDEV5Y4K9wSDII,3388
+emx_onnx_cgen/lowering/attention.py,sha256=19Jq_k0DXwH71a3pmLTWCNMttmw5uuiNK6Jhln5HC4A,16488
+emx_onnx_cgen/lowering/average_pool.py,sha256=9kg3pYHG7QLid_M2dbleC1VoNlVlGsKdOrsWp3pt7sc,8085
+emx_onnx_cgen/lowering/batch_normalization.py,sha256=_aFCm4QaC5jH-JNEvqDFYOyAMdzgUFS_3Gmo1vdPyKE,3987
+emx_onnx_cgen/lowering/cast.py,sha256=zKiE4wI7oWP_TjxBV4fY3-FXvZxK2zy58O6tWJ2dODQ,2852
+emx_onnx_cgen/lowering/common.py,sha256=4w9kjKW3_LilOGgmXYcUGg5lohbYsaLudoL4ALoDUkk,2356
+emx_onnx_cgen/lowering/concat.py,sha256=TefckPfuaIHVHxGExJiO9wlkjyRO1TGg-QAMeoW8hW0,1097
+emx_onnx_cgen/lowering/constant_of_shape.py,sha256=btQflQFMP_y22sK7RrhkbGdaeSSLPC_DWhLjxY7CAgk,3208
+emx_onnx_cgen/lowering/conv.py,sha256=I1_tssw_ySf4beKV0sCVe8DRhNxL58PqC0wxtWjD79s,7309
+emx_onnx_cgen/lowering/cumsum.py,sha256=eX0bDtwY-qevz0KXNHtJaDiKUUHIOhDX0uDiSxcC0ZU,4125
+emx_onnx_cgen/lowering/depth_space.py,sha256=M4md379jiumGWmg7EgR-CinoPzwof2RdfOiNqOzxd9o,4217
+emx_onnx_cgen/lowering/dropout.py,sha256=oBKPMN-J9Gnw8dRXvf-bN15L1-5W7-qKhR72Z6AgLXQ,1775
+emx_onnx_cgen/lowering/elementwise.py,sha256=HN6vEW58lceYECp-7QWLCWOBo1ImyY66aZIg06nA5g8,6231
+emx_onnx_cgen/lowering/expand.py,sha256=4msnYM-6RnzMplQqde2ovOLsjmWQ4bnXEoUiEM6CT6k,5529
+emx_onnx_cgen/lowering/eye_like.py,sha256=76HEdT-EofDCCy7DewjIpILJdIJyJ-YVCbLXO54SX5E,1734
+emx_onnx_cgen/lowering/flatten.py,sha256=sGol05FDN0xoNgSl_DlVbjYvBHCHWjQC2KB15ytYfPs,2142
+emx_onnx_cgen/lowering/gather.py,sha256=9zMB9fcdJi1fkTmDs_-L6FvQi1fnhdk0h7RmeN5MP2M,1814
+emx_onnx_cgen/lowering/gather_elements.py,sha256=0E-WAge15HhGeWPRC_ZE94fb9C6LIoef8p5D1usWvBQ,2358
+emx_onnx_cgen/lowering/gemm.py,sha256=Ps2T4tZgXr5FObz5figwbLZq-Njzg44iBQ9cFmvH78k,4590
+emx_onnx_cgen/lowering/grid_sample.py,sha256=Ne-97ljxSdqfjBJtVHp2AQnEeXGQ5HE-HegCoxcNCm0,5228
+emx_onnx_cgen/lowering/group_normalization.py,sha256=RqXud5_xNxMu8DP5EsPr4L2b6CZghQWCcG6Bh1x2gMA,2664
+emx_onnx_cgen/lowering/identity.py,sha256=fn1Tg56xACwAjhesy0wyr9TJjCmmddnd8QrQ4-uCdO0,1843
+emx_onnx_cgen/lowering/instance_normalization.py,sha256=1Yx2KPKq_BHberCBTrGQXQswAS0FfDle9NpyeD41ypU,1950
+emx_onnx_cgen/lowering/layer_normalization.py,sha256=ZvqGZOhuoYh8ZPyzb-PV0kIc2bbunWTYj12wmrGu9YY,4529
+emx_onnx_cgen/lowering/logsoftmax.py,sha256=1FEaX45GdDr6jIdS_sOwXOy_DdVDruZem4yZ9XA4a38,1669
+emx_onnx_cgen/lowering/lp_normalization.py,sha256=61CGS-2yN0bf5dby5b7Ug1PH3CStZN1xZmYWa5TysTI,1712
+emx_onnx_cgen/lowering/lrn.py,sha256=zGw1Jk7iBk1jHdjdDqfAREsV5VcSdOG3LcAmEllIB08,3370
+emx_onnx_cgen/lowering/lstm.py,sha256=JhGxiF3bTSY3flkw_u9mil2esRxvIjr5Tc4vSPULDr4,12305
+emx_onnx_cgen/lowering/matmul.py,sha256=NEfBa140ofpgm9xnqUBulMSA-yQlb29F2NqhCJpmKSY,4262
+emx_onnx_cgen/lowering/maxpool.py,sha256=MRLeoCEdIwO8JNWOi7iKoeIsJvukqpx_w6GCHaDaYHU,7494
+emx_onnx_cgen/lowering/mean_variance_normalization.py,sha256=L_6ECH9wPEnNX2mL6yroZRexZM8JV5ZnJvoPQS6IAuc,1875
+emx_onnx_cgen/lowering/negative_log_likelihood_loss.py,sha256=wMWvJ9ymkA-ptFgqx0VmHAJHS5NVvDeo2GAlrECztJ8,9307
+emx_onnx_cgen/lowering/pad.py,sha256=cNlh-rA3CRPfO-u8gvJ1MeF1j-vdBMXLuJBpp2DkFTc,10416
+emx_onnx_cgen/lowering/range.py,sha256=xLbG3SGvQiboPqSIh5qZyw3Krbxjk0nd3YvRlUI_q64,3463
+emx_onnx_cgen/lowering/reduce.py,sha256=XRxPopJCU9FGd1XmdxCZickmSTEvmkpAgLUPDFfjRm0,18431
+emx_onnx_cgen/lowering/registry.py,sha256=rKyWnLDBFHJbHw-iyOtXv2Qc5LEBCwgopUXewvQpEpg,1392
+emx_onnx_cgen/lowering/reshape.py,sha256=SF46eP95Z19PT67ayJDKW1fwllBWEZmGfXAL5q9wy-I,6881
+emx_onnx_cgen/lowering/resize.py,sha256=J_x53hVHlfJemLwEhq5n_11Pe1TlF9nRMEpkw6IpzN8,14644
+emx_onnx_cgen/lowering/rms_normalization.py,sha256=_H56Pf9T80FYbmy1m3oc7_D5TbNxRrVeJScD5VmLZRo,2536
+emx_onnx_cgen/lowering/shape.py,sha256=Vvd2zQB06wZcEe4mW5WBRrQuVF8f_tXSM9fpGxe9PEo,2913
+emx_onnx_cgen/lowering/size.py,sha256=Z_DTevdpx2W_3k0GoyQ2uWE3ms_PN1d_Ti7hh6HhB1Q,1261
+emx_onnx_cgen/lowering/slice.py,sha256=yHm_mXeHcLufDmVNvj_kv08zMdbvI39ViHcE-tVPKa0,14816
+emx_onnx_cgen/lowering/softmax.py,sha256=ZaOZf00f5PNHRjSki08Fv-iod6UgqL7cmblfpE_OQRU,1648
+emx_onnx_cgen/lowering/softmax_cross_entropy_loss.py,sha256=I0pbWyJdnf-9vAuX8-xsnovDKDGxynlBhfj5k_IVIa4,5230
+emx_onnx_cgen/lowering/split.py,sha256=ImGnsqrl7IdWbPTPazfXYjcsoRoziqsqtJBum12xTXY,5894
+emx_onnx_cgen/lowering/squeeze.py,sha256=rgICFprcWhC03h-GXZNaIQsdFjsqyxybJYtPklTfaYM,6086
+emx_onnx_cgen/lowering/tile.py,sha256=fT-ybiBZfb4bqBAPrCORZCNm3KWeu4rRW2BJ_UVIVZU,3041
+emx_onnx_cgen/lowering/transpose.py,sha256=TrRXUt-4UFNHZWaOpS3N5zEz5-OCK6-twZdlrnw7Pqg,1762
+emx_onnx_cgen/lowering/unsqueeze.py,sha256=sE3vribz8EyHqDG8lEcreKII7rQDElnHf1OpoM5HiAo,5987
+emx_onnx_cgen/lowering/variadic.py,sha256=hmPzRIj0kcZriGRTR2ma1YMH9g21K_4f-3FXw6qO3jE,3298
+emx_onnx_cgen/lowering/where.py,sha256=uiaWU9RM6o-n38N0AEINIkXS33yVK3-ohkfKIApJOoA,2655
+emx_onnx_cgen/runtime/__init__.py,sha256=88xGpAs1IEBlzlWL_e9tnKUlaSRdc7pQUeVCu5LC4DY,50
+emx_onnx_cgen/runtime/evaluator.py,sha256=GFxrBXcKuQkZ0HY46twOTrNc955UqW3cRKAu5AYVJzQ,84910
+shared/__init__.py,sha256=bmP79AVZdY_1aNULJap9pm76Q41Rabrza6X-0A8lDzw,45
+shared/scalar_functions.py,sha256=OAFO6kT6Gtcv5jp7UBLRhifhGmAbWhDKAmapTvqQruc,89911
+shared/scalar_types.py,sha256=kEpsl5T-NVFxCcTzXqPJbtpvDiCgKHfz91dphLLZxZA,4912
+emx_onnx_cgen-0.2.0.dist-info/METADATA,sha256=xTbPSAdUMfyXwvkxGGVYhsGhKjpxgPSzV1rimuB8zn0,4256
+emx_onnx_cgen-0.2.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+emx_onnx_cgen-0.2.0.dist-info/entry_points.txt,sha256=b7Rvmz_Bi9kWyn7QayQC_FEXiRpt4cS1RnluKh49yoo,57
+emx_onnx_cgen-0.2.0.dist-info/top_level.txt,sha256=g39fo-blEbgiVcC_GRqAnBzN234w3LXbcVdLUoItSLk,21
+emx_onnx_cgen-0.2.0.dist-info/RECORD,,

emx_onnx_cgen-0.2.0.dist-info/WHEEL

@@ -0,0 +1,5 @@
+Wheel-Version: 1.0
+Generator: setuptools (80.9.0)
+Root-Is-Purelib: true
+Tag: py3-none-any
+

emx_onnx_cgen-0.2.0.dist-info/entry_points.txt

@@ -0,0 +1,2 @@
+[console_scripts]
+emx-onnx-cgen = emx_onnx_cgen.cli:main

emx_onnx_cgen-0.2.0.dist-info/top_level.txt

@@ -0,0 +1,2 @@
+emx_onnx_cgen
+shared

shared/__init__.py

@@ -0,0 +1,2 @@
+"""Shared utilities for codegen backend."""
+