Enable PyPI Publishing

.cargo/config.toml

@@ -0,0 +1,2 @@
+[build]
+rustflags = ["-Ctarget-cpu=native"]

.github/workflows/release.yml

@@ -0,0 +1,185 @@
+# This file is autogenerated by maturin v1.12.6
+# To update, run
+#
+#    maturin generate-ci github
+#
+name: Release
+
+on:
+  push:
+    tags: ["v*"]
+  workflow_dispatch:
+  # TODO: Dry run when changing version in arbolta_pyo3/Cargo.toml
+  pull_request:
+
+permissions:
+  contents: read
+
+jobs:
+  linux:
+    runs-on: ${{ matrix.platform.runner }}
+    strategy:
+      matrix:
+        platform:
+          - runner: ubuntu-22.04
+            target: x86_64
+          - runner: ubuntu-22.04
+            target: x86
+          - runner: ubuntu-22.04
+            target: aarch64
+          - runner: ubuntu-22.04
+            target: armv7
+          - runner: ubuntu-22.04
+            target: s390x
+          - runner: ubuntu-22.04
+            target: ppc64le
+    steps:
+      - uses: actions/checkout@v6
+      - uses: actions/setup-python@v6
+        with:
+          python-version: 3.x
+      - name: Build wheels
+        uses: PyO3/maturin-action@v1
+        with:
+          target: ${{ matrix.platform.target }}
+          args: --release --out dist --find-interpreter
+          sccache: ${{ !startsWith(github.ref, 'refs/tags/') }}
+          manylinux: auto
+      - name: Upload wheels
+        uses: actions/upload-artifact@v6
+        with:
+          name: wheels-linux-${{ matrix.platform.target }}
+          path: dist
+
+  musllinux:
+    runs-on: ${{ matrix.platform.runner }}
+    strategy:
+      matrix:
+        platform:
+          - runner: ubuntu-22.04
+            target: x86_64
+          - runner: ubuntu-22.04
+            target: x86
+          - runner: ubuntu-22.04
+            target: aarch64
+          - runner: ubuntu-22.04
+            target: armv7
+    steps:
+      - uses: actions/checkout@v6
+      - uses: actions/setup-python@v6
+        with:
+          python-version: 3.x
+      - name: Build wheels
+        uses: PyO3/maturin-action@v1
+        with:
+          target: ${{ matrix.platform.target }}
+          args: --release --out dist --find-interpreter
+          sccache: ${{ !startsWith(github.ref, 'refs/tags/') }}
+          manylinux: musllinux_1_2
+      - name: Upload wheels
+        uses: actions/upload-artifact@v6
+        with:
+          name: wheels-musllinux-${{ matrix.platform.target }}
+          path: dist
+
+  windows:
+    runs-on: ${{ matrix.platform.runner }}
+    strategy:
+      matrix:
+        platform:
+          - runner: windows-latest
+            target: x64
+            python_arch: x64
+          - runner: windows-latest
+            target: x86
+            python_arch: x86
+          - runner: windows-11-arm
+            target: aarch64
+            python_arch: arm64
+    steps:
+      - uses: actions/checkout@v6
+      - uses: actions/setup-python@v6
+        with:
+          python-version: 3.13
+          architecture: ${{ matrix.platform.python_arch }}
+      - name: Build wheels
+        uses: PyO3/maturin-action@v1
+        with:
+          target: ${{ matrix.platform.target }}
+          args: --release --out dist --find-interpreter
+          sccache: ${{ !startsWith(github.ref, 'refs/tags/') }}
+      - name: Upload wheels
+        uses: actions/upload-artifact@v6
+        with:
+          name: wheels-windows-${{ matrix.platform.target }}
+          path: dist
+
+  macos:
+    runs-on: ${{ matrix.platform.runner }}
+    strategy:
+      matrix:
+        platform:
+          - runner: macos-15-intel
+            target: x86_64
+          - runner: macos-latest
+            target: aarch64
+    steps:
+      - uses: actions/checkout@v6
+      - uses: actions/setup-python@v6
+        with:
+          python-version: 3.x
+      - name: Build wheels
+        uses: PyO3/maturin-action@v1
+        with:
+          target: ${{ matrix.platform.target }}
+          args: --release --out dist --find-interpreter
+          sccache: ${{ !startsWith(github.ref, 'refs/tags/') }}
+      - name: Upload wheels
+        uses: actions/upload-artifact@v6
+        with:
+          name: wheels-macos-${{ matrix.platform.target }}
+          path: dist
+
+  sdist:
+    runs-on: ubuntu-latest
+    steps:
+      - uses: actions/checkout@v6
+      - name: Build sdist
+        uses: PyO3/maturin-action@v1
+        with:
+          command: sdist
+          args: --out dist
+      - name: Upload sdist
+        uses: actions/upload-artifact@v6
+        with:
+          name: wheels-sdist
+          path: dist
+
+  release:
+    name: Publish to PyPI
+    environment:
+      name: pypi
+    runs-on: ubuntu-latest
+    if: ${{ startsWith(github.ref, 'refs/tags/') || github.event_name == 'workflow_dispatch' }}
+    needs: [linux, musllinux, windows, macos, sdist]
+    permissions:
+      # Use to sign the release artifacts
+      id-token: write
+      # Used to upload release artifacts
+      contents: write
+      # Used to generate artifact attestation
+      attestations: write
+    steps:
+      - uses: actions/download-artifact@v7
+      - name: Generate artifact attestation
+        uses: actions/attest-build-provenance@v3
+        with:
+          subject-path: 'wheels-*/*'
+      - name: Install uv
+        if: ${{ startsWith(github.ref, 'refs/tags/') }}
+        uses: astral-sh/setup-uv@v7
+      - name: Publish to PyPI
+        if: ${{ startsWith(github.ref, 'refs/tags/') }}
+        run: uv publish 'wheels-*/*'
+        env:
+          UV_PUBLISH_TOKEN: ${{ secrets.PYPI_API_TOKEN }}

.gitignore

@@ -1,4 +1,5 @@
 ### Python ###
+uv.lock
 # Byte-compiled / optimized / DLL files
 __pycache__/
 *.py[cod]
@@ -157,4 +158,4 @@ Cargo.lock
 
 ### Generated ###
 examples/notebooks/*/output
-synth_output
+*.ruff_cache

.pre-commit-config.yaml

@@ -1,5 +1,3 @@
-exclude: '(examples/pre_synthesized_demo/i8bxi8bx16_vector_mac.json)'
-
 repos:
   - repo: https://github.com/pre-commit/pre-commit-hooks
     rev: v5.0.0
@@ -14,26 +12,30 @@ repos:
       - id: check-json
       - id: check-added-large-files
       - id: requirements-txt-fixer
-  - repo: https://github.com/doublify/pre-commit-rust
-    rev: v1.0
+  - repo: local
     hooks:
-      - id: fmt
-        name: fmt (rust)
-      - id: clippy
-        name: clippy (rust)
+      - id: cargo-fmt
+        name: cargo fmt
+        entry: cargo fmt --all -- --check
+        language: system
+        types: [rust]
+      - id: cargo-clippy
+        name: cargo clippy
+        entry: cargo clippy --workspace --all-targets -- -Dclippy::all
+        pass_filenames: false
+        language: system
+        types: [rust]
       - id: cargo-check
-        name: cargo-check (rust)
-  - repo: https://github.com/pycqa/isort
-    rev: 5.13.2
+        name: cargo check
+        entry: cargo check --workspace --all-targets
+        pass_filenames: false
+        language: system
+        types: [rust]
+  - repo: https://github.com/astral-sh/ruff-pre-commit
+    rev: v0.11.8
     hooks:
-      - id: isort
-        name: isort (python)
-  - repo: https://github.com/google/yapf
-    rev: v0.43.0
-    hooks:
-      - id: yapf
-        name: yapf (python)
-        args: ["-i"]
+      - id: ruff
+      - id: ruff-format
   - repo: https://github.com/srstevenson/nb-clean
     rev: 4.0.1
     hooks:

Cargo.toml

@@ -1,3 +1,32 @@
 [workspace]
 members = ["crates/*"]
-resolver = "2"
+resolver = "3"
+
+[workspace.package]
+edition = "2024"
+repository = "https://github.com/KastnerRG/arbolta"
+homepage = "https://github.com/KastnerRG/arbolta"
+authors = ["Alexander Redding <alredding@ucsd.edu>"]
+license = "MIT"
+
+[workspace.dependencies]
+anyhow = "1"
+arbolta = { path = "crates/arbolta" }
+derive_more = { version = "2", features = ["full"] }
+enum_dispatch = "0.3"
+indexmap = "2.12"
+ndarray = "0.16.1"
+num-traits = "0.2"
+once_cell = "1.21"
+petgraph = { version = "0.8.3", features = ["serde-1"] }
+postcard = "1"
+rstest = "0.26"
+serde = { version = "1", features = ["derive"] }
+thiserror = "2.0"
+yosys-netlist-json = { git = "https://github.com/alexredd99/yosys-netlist-json.git" }
+
+[profile.release]
+lto = true
+codegen-units = 1
+opt-level = 3
+debug = false

LICENSE

@@ -1,6 +1,7 @@
 MIT License
 
-Copyright (c) 2024 - 2025 Advanced Micro Devices, Inc. All rights reserved.
+Copyright (c) 2024 - 2025 Advanced Micro Devices, Inc.
+Copyright (c) 2026 Alexander Redding
 
 Permission is hereby granted, free of charge, to any person obtaining a copy
 of this software and associated documentation files (the "Software"), to deal

crates/arbolta/Cargo.toml

@@ -1,15 +1,31 @@
 [package]
 name = "arbolta"
 version = "0.1.0"
-authors = ["AMD Research & Advanced Development"]
-edition = "2021"
+authors = { workspace = true }
+edition = { workspace = true }
+repository = { workspace = true }
+homepage = { workspace = true }
+license = { workspace = true }
+description = "Core Arbolta hardware simulator"
+exclude = [
+    "tests/*"
+]
 
 [dependencies]
-serde = { version = "1.0", features = ["derive"] }
-ndarray = "0.16.1"
-num-traits = "0.2"
-once_cell = "1.19.0"
-rstest = "0.23.0"
-flexbuffers = "2.0.0"
-yosys-netlist-json = { git = "https://github.com/alexredd99/yosys-netlist-json.git" }
-thiserror = "2.0.3"
+anyhow = { workspace = true }
+derive-new = "0.7"
+derive_more = { workspace = true }
+enum_dispatch = { workspace = true }
+indexmap = { workspace = true }
+inventory = "0.3.21"
+ndarray = { workspace = true }
+num-traits = { workspace = true }
+once_cell = { workspace = true }
+paste = "1.0.15"
+petgraph = { workspace = true }
+postcard = { workspace = true }
+rstest = { workspace = true }
+serde = { workspace = true }
+serde_json = "1.0.145"
+thiserror = { workspace = true }
+yosys-netlist-json = { workspace = true }