trntensor 0.1.2__tar.gz → 0.17.0__tar.gz

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

@@ -0,0 +1,59 @@
+name: CI
+
+on:
+  push:
+    branches: [main]
+  pull_request:
+
+# Opt into Node.js 24 for all third-party JavaScript actions ahead of
+# GitHub's June 2026 default switch. Silences the Node 20 deprecation
+# notice emitted by setup-uv and codecov-action's github-script dep.
+env:
+  FORCE_JAVASCRIPT_ACTIONS_TO_NODE24: true
+
+jobs:
+  lint:
+    runs-on: ubuntu-latest
+    steps:
+      - uses: actions/checkout@v6
+      - uses: astral-sh/setup-uv@v6
+      - run: uvx ruff check .
+      - run: uvx ruff format --check .
+
+  test:
+    runs-on: ubuntu-latest
+    strategy:
+      matrix:
+        python-version: ["3.10", "3.11", "3.12"]
+    steps:
+      - uses: actions/checkout@v6
+      - uses: astral-sh/setup-uv@v6
+        with:
+          python-version: ${{ matrix.python-version }}
+      - run: uv sync --extra dev
+      - run: uv run pytest tests/ -v -x --tb=short -m "not neuron and not nki_simulator" --cov=trntensor --cov-report=xml
+
+      - name: Upload coverage reports to Codecov
+        uses: codecov/codecov-action@v5
+        with:
+          token: ${{ secrets.CODECOV_TOKEN }}
+          slug: trnsci/trntensor
+
+  nki-simulator:
+    # 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).
+    runs-on: ubuntu-latest
+    steps:
+      - uses: actions/checkout@v6
+      - uses: astral-sh/setup-uv@v6
+        with:
+          python-version: "3.12"
+      - run: uv sync --extra dev
+      - name: Install NKI simulator
+        run: uv pip install --extra-index-url https://pip.repos.neuron.amazonaws.com "nki>=0.3.0"
+      - name: Run simulator-backed kernel tests
+        env:
+          TRNTENSOR_USE_SIMULATOR: "1"
+        run: uv run pytest tests/ -v -m nki_simulator --tb=short

{trntensor-0.1.2 → trntensor-0.17.0}/.github/workflows/notify-umbrella.yml

@@ -18,6 +18,9 @@ on:
       - "CHANGELOG.md"
   workflow_dispatch:
 
+env:
+  FORCE_JAVASCRIPT_ACTIONS_TO_NODE24: true
+
 jobs:
   dispatch:
     runs-on: ubuntu-latest

{trntensor-0.1.2 → trntensor-0.17.0}/.github/workflows/publish.yml

@@ -4,6 +4,9 @@ on:
   release:
     types: [published]
 
+env:
+  FORCE_JAVASCRIPT_ACTIONS_TO_NODE24: true
+
 jobs:
   build:
     runs-on: ubuntu-latest

{trntensor-0.1.2 → trntensor-0.17.0}/.gitignore

@@ -31,3 +31,9 @@ site/
 # coverage artifacts
 .coverage
 htmlcov/
+
+# terraform local state
+infra/terraform/.terraform/
+infra/terraform/terraform.tfstate*
+infra/terraform/.terraform.lock.hcl
+benchmark_results.json

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

@@ -0,0 +1,11 @@
+# Install locally: pip install pre-commit && pre-commit install
+# Runs on `git commit`; ruff auto-fixes staged Python files then formats them.
+# CI re-checks the same rules in the lint job — pre-commit keeps CI green.
+
+repos:
+  - repo: https://github.com/astral-sh/ruff-pre-commit
+    rev: v0.6.9
+    hooks:
+      - id: ruff
+        args: [--fix]
+      - id: ruff-format