trnsolver 0.3.0__tar.gz → 0.17.0__tar.gz

trnsolver-0.17.0/.github/workflows/ci.yml

@@ -0,0 +1,109 @@
+name: CI
+
+on:
+  push:
+    branches: [main]
+  pull_request:
+
+jobs:
+  lint:
+    runs-on: ubuntu-latest
+    steps:
+      - uses: actions/checkout@v6
+      - uses: actions/setup-python@v6
+        with:
+          python-version: "3.12"
+      - uses: astral-sh/setup-uv@v5
+        with:
+          enable-cache: true
+      - run: uv pip install --system "ruff>=0.6"
+      - run: ruff check .
+      - run: ruff format --check .
+
+  test:
+    runs-on: ubuntu-latest
+    strategy:
+      matrix:
+        python-version: ["3.10", "3.11", "3.12"]
+    steps:
+      - uses: actions/checkout@v6
+      - uses: actions/setup-python@v6
+        with:
+          python-version: ${{ matrix.python-version }}
+      - uses: astral-sh/setup-uv@v5
+        with:
+          enable-cache: true
+      - run: uv pip install --system -e ".[dev]"
+      - run: pytest tests/ -v -x --tb=short -m "not neuron and not nki_simulator" --cov=trnsolver --cov-report=xml
+      - name: Upload coverage reports to Codecov
+        uses: codecov/codecov-action@v5
+        with:
+          token: ${{ secrets.CODECOV_TOKEN }}
+          slug: trnsci/trnsolver
+
+  # Runs NKI kernels through nki.simulate(kernel)(numpy_args) on CPU.
+  # Catches Python-trace-level errors (bad kwargs, dropped ops, shape
+  # mismatches) pre-merge without AWS round-trips. MLIR verifier errors
+  # remain hardware-only (simulator explicitly skips compile). NKI 0.3.0
+  # Stable is the canonical namespace — package is `nki`, not `neuronxcc.nki`.
+  nki-simulator:
+    runs-on: ubuntu-latest
+    steps:
+      - uses: actions/checkout@v6
+      - uses: actions/setup-python@v6
+        with:
+          python-version: "3.12"
+      - uses: astral-sh/setup-uv@v5
+        with:
+          enable-cache: true
+      - name: Install trnsolver + NKI simulator deps
+        # --index-strategy unsafe-best-match: uv normally locks a package to
+        # the first index it's found on; nki's transitive deps (e.g. islpy)
+        # exist on both the Neuron repo and PyPI. Match pip's multi-index
+        # behavior so resolution picks the newest available wheel per package.
+        run: |
+          uv pip install --system -e ".[dev]"
+          uv pip install --system --index-strategy unsafe-best-match "nki>=0.3.0" --extra-index-url https://pip.repos.neuron.amazonaws.com
+      - name: Run simulator-backed kernel tests
+        env:
+          TRNSOLVER_USE_SIMULATOR: "1"
+        run: pytest tests/ -v -m nki_simulator --tb=short --cov=trnsolver --cov-report=xml --cov-append
+      - name: Upload coverage reports to Codecov
+        uses: codecov/codecov-action@v5
+        with:
+          token: ${{ secrets.CODECOV_TOKEN }}
+          slug: trnsci/trnsolver
+          flags: simulator
+
+  # pytest-benchmark bit-rot gate. Runs the benchmark suite with minimum
+  # rounds on ubuntu-latest to catch import / fixture / API drift before
+  # it hits the real perf measurement path. No perf gate — just execution.
+  # Real perf numbers come from the hardware (`run_neuron_tests.sh`) and
+  # g5 (`run_cuda_tests.sh`) runners.
+  bench-smoke:
+    runs-on: ubuntu-latest
+    steps:
+      - uses: actions/checkout@v6
+      - uses: actions/setup-python@v6
+        with:
+          python-version: "3.12"
+      - uses: astral-sh/setup-uv@v5
+        with:
+          enable-cache: true
+      - run: uv pip install --system -e ".[dev]"
+      - name: Run benchmark smoke (exclude neuron + cuda)
+        # --override-ini: bench_*.py is outside the test_*.py default pattern;
+        # override rather than pollute the pyproject test config.
+        run: |
+          pytest benchmarks/ -v --benchmark-only -m "not neuron and not cuda" \
+            --override-ini 'python_files=bench_*.py' \
+            --benchmark-min-rounds=2 \
+            --benchmark-json=bench.json \
+            --tb=short
+      - name: Upload benchmark JSON
+        if: always()
+        uses: actions/upload-artifact@v4
+        with:
+          name: bench-smoke-results
+          path: bench.json
+          retention-days: 14

{trnsolver-0.3.0 → trnsolver-0.17.0}/.github/workflows/publish.yml

@@ -12,7 +12,10 @@ jobs:
       - uses: actions/setup-python@v6
         with:
           python-version: "3.12"
-      - run: pip install build
+      - uses: astral-sh/setup-uv@v5
+        with:
+          enable-cache: true
+      - run: uv pip install --system build
       - run: python -m build
       - uses: actions/upload-artifact@v4
         with:

{trnsolver-0.3.0 → trnsolver-0.17.0}/.gitignore

@@ -7,6 +7,9 @@ build/
 .eggs/
 *.egg
 .pytest_cache/
+.coverage
+coverage.xml
+htmlcov/
 .tox/
 .venv/
 venv/

trnsolver-0.17.0/.pre-commit-config.yaml

@@ -0,0 +1,31 @@
+# Pre-commit hooks for trnsolver.
+#
+# Setup (once per clone):
+#   pip install pre-commit
+#   pre-commit install
+#
+# After that, `git commit` runs ruff on staged files automatically. CI enforces
+# the same checks (`.github/workflows/ci.yml::lint`), so skipping hooks locally
+# just moves the failure to CI.
+#
+# Run ad-hoc against all files:
+#   pre-commit run --all-files
+
+repos:
+  - repo: https://github.com/astral-sh/ruff-pre-commit
+    rev: v0.7.4
+    hooks:
+      - id: ruff
+        args: [--fix]
+      - id: ruff-format
+
+  - repo: https://github.com/pre-commit/pre-commit-hooks
+    rev: v5.0.0
+    hooks:
+      - id: trailing-whitespace
+      - id: end-of-file-fixer
+      - id: check-merge-conflict
+      - id: check-yaml
+      - id: check-toml
+      - id: check-added-large-files
+        args: [--maxkb=500]