From 866a33943eba3e3de64b4f8e32c200e3a52f619b Mon Sep 17 00:00:00 2001 From: Hiroshi Shinaoka Date: Mon, 16 Mar 2026 08:44:02 +0900 Subject: [PATCH 1/4] docs: add complex support ledger plan --- ...026-03-16-complex-support-ledger-design.md | 321 ++++++++++++++ ...6-complex-support-ledger-implementation.md | 398 ++++++++++++++++++ 2 files changed, 719 insertions(+) create mode 100644 docs/plans/2026-03-16-complex-support-ledger-design.md create mode 100644 docs/plans/2026-03-16-complex-support-ledger-implementation.md diff --git a/docs/plans/2026-03-16-complex-support-ledger-design.md b/docs/plans/2026-03-16-complex-support-ledger-design.md new file mode 100644 index 0000000..197ee1e --- /dev/null +++ b/docs/plans/2026-03-16-complex-support-ledger-design.md @@ -0,0 +1,321 @@ +# Complex Support Ledger Design + +## Goal + +Make this repository the machine-readable source of truth for complex AD +capability on the full published `(op, family)` surface before downstream +`tenferro-rs` implementation work proceeds. + +The repository should be able to answer, from checked-in artifacts and CI +alone: + +- which families have a complex-reviewed math note +- which families have full checked-in complex DB coverage +- which families are explicitly unsupported for complex and why + +## Problem + +The repository already has two partial sources of truth: + +- `docs/math/registry.json` answers where published DB families map into math + notes +- `cases/*/*.jsonl` answers which concrete records are currently materialized + +That is not enough for issue `#14`. + +The missing piece is an explicit ledger for complex readiness. Without it: + +- the issue checklist is the only place that tries to summarize readiness +- CI cannot decide whether a family is reviewed, covered, unsupported, or + simply forgotten +- downstream replay consumers cannot distinguish "not supported" from "not yet + audited" +- the current publish-coverage report can show dtype presence, but not whether + a family is mathematically reviewed for complex mode + +## Design Principles + +1. Keep the source of truth inside the repository. +2. Preserve single responsibility: + - `registry.json` keeps note-location linkage only + - the new ledger tracks complex capability state +3. Track status at exact `(op, family)` granularity. +4. Derive readiness from explicit note and DB states rather than storing a + redundant `ready` bit. +5. Generate human-facing reports from the ledger; do not maintain large status + tables by hand. +6. Reuse the current repo surface as the canonical family universe instead of + manually duplicating the issue checklist in code. + +## Upstream Review Inputs + +The note audit for this work should use the local PyTorch checkout at +`../pytorch` as the upstream reference corpus alongside the pinned runtime +contract in this repository. + +Relevant local upstream files include: + +- `../pytorch/torch/csrc/autograd/FunctionsManual.cpp` +- `../pytorch/tools/autograd/derivatives.yaml` +- `../pytorch/torch/testing/_internal/opinfo/definitions/linalg.py` +- `../pytorch/torch/testing/_internal/common_methods_invocations.py` +- `../pytorch/test/test_linalg.py` + +These paths provide the complex manual formulas, wrapper conventions, OpInfo +family inventory, and gauge-related failure behavior that the local math notes +and ledger statuses need to reflect. + +## Scope + +### Included + +- Every currently published `(op, family)` in + `generators.pytorch_v1.build_case_spec_index()` +- All existing math-note linkage in `docs/math/registry.json` +- A new complex-support ledger under `docs/math/` +- A validator script and tests for ledger integrity +- A generated human-facing complex-support report under `docs/generated/` +- Math-note updates where the complex audit shows that existing prose is still + incomplete for a family we intend to mark reviewed +- Explicit unsupported classification for float-only or intentionally deferred + complex families + +### Excluded + +- Any schema change to the JSONL case format +- Any attempt to merge the complex ledger into `docs/math/registry.json` +- GitHub issue text as a machine-readable source of truth +- New downstream replay behavior inside `tenferro-rs` + +## Canonical Ledger + +Add a new checked-in artifact: + +- `docs/math/complex-support.json` + +The canonical shape is: + +```json +{ + "version": 1, + "entries": [ + { + "op": "svd", + "family": "u_abs", + "note": { + "path": "docs/math/svd.md", + "anchor": "family-u-abs", + "status": "reviewed" + }, + "db": { + "status": "covered" + }, + "unsupported_reason": null + } + ] +} +``` + +Each entry represents exactly one `(op, family)` pair. + +### Note Axis + +`note.status` is one of: + +- `reviewed` +- `pending` +- `not_required` + +Rules: + +- `reviewed` requires non-null `note.path` and `note.anchor` +- `pending` also carries the intended `note.path` and `note.anchor` when a note + target already exists +- `not_required` requires `note.path = null` and `note.anchor = null` + +`not_required` is for families whose complex capability depends on an existing +shared or adjacent rule but does not require a distinct complex note-review +decision at this family boundary, for example wrapper-style DB-only families. + +### DB Axis + +`db.status` is one of: + +- `covered` +- `pending` +- `unsupported` + +Rules: + +- `covered` means all publishable complex dtypes for this family are present in + the checked-in DB +- `pending` means the family is intended to be supported but the checked-in DB + is not yet complete +- `unsupported` means the family is intentionally out of current complex scope + +### Unsupported Reason + +`unsupported_reason` is required iff `db.status == "unsupported"`. + +It is not a free-floating override; it is the required justification for an +unsupported DB decision. + +## Derived Semantics + +The ledger does not store a separate readiness field. + +Downstream-ready status is derived as: + +```text +note.status == reviewed && db.status == covered +``` + +Unsupported status is derived as: + +```text +db.status == unsupported && unsupported_reason is non-empty +``` + +Everything else is incomplete. + +## Relationship To Existing Artifacts + +### `docs/math/registry.json` + +The registry remains the source of truth for mapping published DB families to +note locations and anchors. + +It continues to answer: + +- where the note lives +- which stable anchor corresponds to a DB family + +It does not answer: + +- whether the family was complex-reviewed +- whether the DB has complex coverage +- whether the family is intentionally unsupported + +### `cases/*/*.jsonl` + +The checked-in JSONL case tree remains the source of truth for what numeric +coverage actually exists. + +The ledger does not duplicate published dtype payloads. Instead, the validator +derives complex coverage from the case tree and checks that against the ledger's +declared `db.status`. + +## Validation And CI + +Add a dedicated validator module and script: + +- `validators/complex_support.py` +- `scripts/check_complex_support.py` + +Validation should enforce: + +- the ledger exists and has versioned structure +- every repo-tracked `(op, family)` appears exactly once +- there are no duplicate entries +- reviewed note targets exist on disk and expose the declared anchor +- reviewed note targets match `docs/math/registry.json` when a registry entry is + present for that family +- `covered` DB entries have all publishable complex dtypes materialized in the + checked-in case tree +- `unsupported` DB entries have a non-empty reason +- checked-in repository state contains no `pending` entries once this issue is + completed + +The canonical family universe for these checks is +`generators.pytorch_v1.build_case_spec_index()`. This keeps the validator tied +to the same surface the repository already publishes and verifies elsewhere. + +## Generated Report + +Add a generated report: + +- `docs/generated/complex-support.md` + +Generated from: + +- `docs/math/complex-support.json` +- `docs/math/registry.json` +- `generators.pytorch_v1.build_case_spec_index()` +- the checked-in `cases/` tree + +The report should include: + +- summary counts for ready, unsupported, and pending families +- a full flat table of family status +- focused sections for ready, unsupported, and pending subsets +- published complex dtype coverage per family + +The report is the human-facing artifact. The JSON ledger is the machine-readable +artifact. + +## Note Audit Strategy + +The math-note audit should be structured around the existing shared-note layout: + +- dedicated linalg notes such as `svd.md`, `qr.md`, `lu.md`, `solve.md`, + `cholesky.md`, `inv.md`, `det.md`, `eig.md`, `eigen.md`, `lstsq.md`, + `matrix_exp.md`, `norm.md`, and `pinv.md` +- shared scalar/wrapper note `scalar_ops.md` + +Families should only be marked `reviewed` when the checked-in note is adequate +for complex mode relative to the local upstream references in `../pytorch`. +Otherwise: + +- update the note and then mark `reviewed`, or +- keep `pending`, or +- mark DB support `unsupported` with a reason when complex support is not + intended yet + +## Expected Initial Audit Outcome + +Current repo state already appears to have: + +- full registry coverage for all published `(op, family)` pairs +- full checked-in complex DB coverage for the complex-publishable family set +- a large set of scalar and linalg notes that are already explicitly + complex-aware + +This suggests that the main missing artifact is the explicit complex-support +ledger and its generated report, plus any residual note edits discovered during +the manual review. + +Float-only published families should generally end up as explicit complex +`unsupported` entries rather than appearing silent or pending forever. + +## Risks + +### Risk: conflating generic note linkage with complex review state + +Mitigation: + +- keep the ledger separate from `registry.json` +- validate only reviewed entries against the registry + +### Risk: calling a family "covered" when only partial complex dtype support is present + +Mitigation: + +- derive coverage from the case tree and current generator spec +- require all publishable complex dtypes for `covered` + +### Risk: audit drift between notes, ledger, and report + +Mitigation: + +- generate the report from the ledger +- validate note targets and DB coverage in CI + +## Deliverables + +1. `docs/math/complex-support.json` +2. `validators/complex_support.py` +3. `scripts/check_complex_support.py` +4. `scripts/report_complex_support.py` +5. `docs/generated/complex-support.md` +6. Tests covering validator and report behavior +7. Any required math-note updates discovered during the complex audit diff --git a/docs/plans/2026-03-16-complex-support-ledger-implementation.md b/docs/plans/2026-03-16-complex-support-ledger-implementation.md new file mode 100644 index 0000000..b371148 --- /dev/null +++ b/docs/plans/2026-03-16-complex-support-ledger-implementation.md @@ -0,0 +1,398 @@ +# Complex Support Ledger Implementation Plan + +> **For Claude:** REQUIRED SUB-SKILL: Use superpowers:executing-plans to implement this plan task-by-task. + +**Goal:** Add a machine-readable complex-support ledger, validate it against the published DB and math-note registry, generate a human-facing complex-support report, and complete the remaining note/unsupported classifications needed for issue `#14`. + +**Architecture:** Keep `docs/math/registry.json` responsible only for note linkage and add a separate `docs/math/complex-support.json` ledger at exact `(op, family)` granularity. Validate the ledger against the current published family surface from `generators.pytorch_v1.build_case_spec_index()`, the checked-in `cases/` tree, and note anchors on disk. Generate `docs/generated/complex-support.md` from that ledger and finish the note/unsupported audit using the local `../pytorch` tree as the upstream review reference. + +**Tech Stack:** Python 3.12, `uv`, pinned `torch==2.10.0`, `unittest`, JSON, Markdown generation, existing validators/scripts/test patterns, local upstream references under `../pytorch`. + +--- + +### Task 1: Add ledger-validator unit tests + +**Files:** +- Create: `tests/test_complex_support.py` +- Modify: `tests/test_scripts.py` + +**Step 1: Write the failing test** + +Add tests that require a complex-support validator to reject: + +- duplicate `(op, family)` entries +- missing reviewed note paths or anchors +- `not_required` entries with non-null note metadata +- `unsupported` DB entries without a reason +- `covered` entries whose case tree does not include the expected complex dtypes + +Use a temporary repository fixture similar to `tests/test_math_registry.py`, +with a tiny injected spec surface such as: + +```python +spec_index = { + ("solve", "identity"): StubSpec( + op="solve", + family="identity", + supported_dtype_names=("float64", "complex128"), + ) +} +``` + +Also add one script test asserting that `scripts/check_complex_support.py` +exits successfully on the checked-in repo once implemented. + +**Step 2: Run test to verify it fails** + +Run: + +```bash +uv run python -m unittest tests.test_complex_support tests.test_scripts -v +``` + +Expected: FAIL because no complex-support validator or script exists yet. + +**Step 3: Write minimal implementation** + +Create: + +- `validators/complex_support.py` +- `scripts/check_complex_support.py` + +Implement helpers to: + +- load the ledger +- compute published complex dtypes from `cases/*/*.jsonl` +- validate note/db invariants against an injected or default spec surface + +Keep the API close to `validators/math_registry.py` so the tests can use +temporary repos without mocking the full application. + +**Step 4: Run test to verify it passes** + +Run: + +```bash +uv run python -m unittest tests.test_complex_support tests.test_scripts -v +``` + +Expected: PASS + +**Step 5: Commit** + +```bash +git add tests/test_complex_support.py tests/test_scripts.py validators/complex_support.py scripts/check_complex_support.py +git commit -m "feat: validate complex support ledger" +``` + +### Task 2: Add the checked-in ledger and repo-level classification tests + +**Files:** +- Create: `docs/math/complex-support.json` +- Modify: `tests/test_complex_support.py` +- Modify: `README.md` + +**Step 1: Write the failing test** + +Add repo-level tests that require: + +- the checked-in ledger to exist +- every current `(op, family)` from `build_case_spec_index()` to appear exactly once +- representative complex-covered families such as `svd/u_abs`, `solve/identity`, + `eig/values_vectors_abs`, and `sum/identity` to be marked `reviewed + covered` +- representative DB-only note cases such as `lu_factor_ex/identity`, + `solve_ex/identity`, and `inv_ex/identity` to be marked `note.status == "not_required"` +- representative float-only families such as `atan2/identity`, + `copysign/identity`, and `special_ndtr/identity` to be marked + `db.status == "unsupported"` with a reason + +Document the new ledger in `README.md`. + +**Step 2: Run test to verify it fails** + +Run: + +```bash +uv run python -m unittest tests.test_complex_support tests.test_repo_config -v +``` + +Expected: FAIL because the ledger file is not checked in yet and the README does +not mention it. + +**Step 3: Write minimal implementation** + +Add `docs/math/complex-support.json` with one flat entry per published +`(op, family)`: + +- `reviewed + covered` for families whose math note and checked-in DB are ready +- `not_required + covered` for DB-only wrapper families that do not need a + separate complex note review +- `reviewed + unsupported` or `not_required + unsupported` for float-only or + intentionally deferred complex families + +The file should contain no `pending` entries in the final checked-in state. + +Update `README.md` to document: + +- the new ledger path +- the validation script +- the generated complex-support report + +**Step 4: Run test to verify it passes** + +Run: + +```bash +uv run python -m unittest tests.test_complex_support tests.test_repo_config -v +``` + +Expected: PASS + +**Step 5: Commit** + +```bash +git add docs/math/complex-support.json tests/test_complex_support.py README.md +git commit -m "docs: add checked-in complex support ledger" +``` + +### Task 3: Add report-generation tests and script + +**Files:** +- Create: `tests/test_complex_support_report.py` +- Create: `scripts/report_complex_support.py` +- Create: `docs/generated/complex-support.md` + +**Step 1: Write the failing test** + +Add tests that require the report generator to: + +- emit a Markdown report with summary counts +- include a full table with `op`, `family`, `note status`, `db status`, + `complex published dtypes`, `unsupported reason`, and `ready` +- include representative rows such as: + +```text +| svd | u_abs | reviewed | covered | complex128, complex64 | - | yes | +``` + +- include representative unsupported rows such as `atan2/identity` +- reproduce the checked-in `docs/generated/complex-support.md` + +**Step 2: Run test to verify it fails** + +Run: + +```bash +uv run python -m unittest tests.test_complex_support_report -v +``` + +Expected: FAIL because the report script and checked-in report do not exist yet. + +**Step 3: Write minimal implementation** + +Create `scripts/report_complex_support.py` that: + +- loads `docs/math/complex-support.json` +- loads the current DB/spec surface +- computes published complex dtypes from the case tree +- emits `docs/generated/complex-support.md` + +Check in the generated report. + +**Step 4: Run test to verify it passes** + +Run: + +```bash +uv run python -m unittest tests.test_complex_support_report -v +``` + +Expected: PASS + +**Step 5: Commit** + +```bash +git add tests/test_complex_support_report.py scripts/report_complex_support.py docs/generated/complex-support.md +git commit -m "docs: add complex support report" +``` + +### Task 4: Audit note coverage and lock representative reviewed classifications + +**Files:** +- Modify: `docs/math/scalar_ops.md` +- Modify: `docs/math/svd.md` +- Modify: `docs/math/qr.md` +- Modify: `docs/math/lu.md` +- Modify: `docs/math/solve.md` +- Modify: `docs/math/cholesky.md` +- Modify: `docs/math/inv.md` +- Modify: `docs/math/det.md` +- Modify: `docs/math/eig.md` +- Modify: `docs/math/eigen.md` +- Modify: `docs/math/lstsq.md` +- Modify: `docs/math/matrix_exp.md` +- Modify: `docs/math/norm.md` +- Modify: `docs/math/pinv.md` +- Modify: `docs/math/complex-support.json` +- Modify: `tests/test_math_registry.py` +- Modify: `tests/test_complex_support.py` + +**Step 1: Write the failing test** + +Add targeted tests that pin representative complex-reviewed note coverage: + +- `eig.md` still mentions normalization correction and gauge invariance +- `eigen.md` still documents the Hermitian complex rule +- `svd.md`, `solve.md`, `pinv.md`, and `norm.md` still retain the key complex + derivation markers relied on by the audit +- representative ledger entries for those families are marked + `note.status == "reviewed"` + +For any note found incomplete during the audit, first add an assertion for the +missing complex detail before editing the note. + +**Step 2: Run test to verify it fails** + +Run: + +```bash +uv run python -m unittest tests.test_math_registry tests.test_complex_support -v +``` + +Expected: FAIL on the first audited family whose note-review status or note +content is still incomplete for complex mode. + +**Step 3: Write minimal implementation** + +Audit the shared scalar note and the dedicated linalg notes against the local +upstream references under `../pytorch`, then: + +- strengthen any note whose complex explanation is still missing +- update the ledger from provisional classifications to final `reviewed` or + `not_required` values + +Do not create new dedicated note files unless the audit actually needs them; +prefer strengthening the current shared-note layout first. + +**Step 4: Run test to verify it passes** + +Run: + +```bash +uv run python -m unittest tests.test_math_registry tests.test_complex_support -v +``` + +Expected: PASS + +**Step 5: Commit** + +```bash +git add docs/math/scalar_ops.md docs/math/svd.md docs/math/qr.md docs/math/lu.md docs/math/solve.md docs/math/cholesky.md docs/math/inv.md docs/math/det.md docs/math/eig.md docs/math/eigen.md docs/math/lstsq.md docs/math/matrix_exp.md docs/math/norm.md docs/math/pinv.md docs/math/complex-support.json tests/test_math_registry.py tests/test_complex_support.py +git commit -m "docs: audit complex note coverage" +``` + +### Task 5: Wire the new checks into repository verification + +**Files:** +- Modify: `tests/test_scripts.py` +- Modify: `tests/test_repo_config.py` +- Modify: `README.md` +- Modify: `.github/workflows/oracle-integrity.yml` + +**Step 1: Write the failing test** + +Add tests that require repository verification to mention and run the new +ledger/report checks: + +- `README.md` documents `uv run python scripts/check_complex_support.py` +- `README.md` documents `uv run python scripts/report_complex_support.py` +- `.github/workflows/oracle-integrity.yml` includes the complex-support check + +**Step 2: Run test to verify it fails** + +Run: + +```bash +uv run python -m unittest tests.test_repo_config tests.test_scripts -v +``` + +Expected: FAIL because the README and workflow do not yet mention the new check. + +**Step 3: Write minimal implementation** + +Update: + +- `README.md` +- `.github/workflows/oracle-integrity.yml` + +so the repository contract includes the new validator and report artifact. + +**Step 4: Run test to verify it passes** + +Run: + +```bash +uv run python -m unittest tests.test_repo_config tests.test_scripts -v +``` + +Expected: PASS + +**Step 5: Commit** + +```bash +git add tests/test_scripts.py tests/test_repo_config.py README.md .github/workflows/oracle-integrity.yml +git commit -m "ci: enforce complex support checks" +``` + +### Task 6: Full verification and checked-artifact regeneration + +**Files:** +- Modify: `docs/generated/complex-support.md` +- Modify: `docs/math/complex-support.json` +- Modify: any files touched in Tasks 1-5 + +**Step 1: Run the focused tests** + +Run: + +```bash +uv run python -m unittest tests.test_complex_support tests.test_complex_support_report tests.test_math_registry tests.test_scripts tests.test_repo_config -v +``` + +Expected: PASS + +**Step 2: Run the ledger and docs checks** + +Run: + +```bash +uv run python scripts/check_complex_support.py +uv run python scripts/check_math_registry.py +uv run python scripts/report_complex_support.py +``` + +Expected: + +- `complex_support_ok=1` +- `math_registry_ok=1` +- `complex_support_report=.../docs/generated/complex-support.md` + +**Step 3: Run repository contract verification** + +Run: + +```bash +uv run python scripts/validate_schema.py +uv run python scripts/verify_cases.py +uv run python scripts/check_replay.py +uv run python scripts/check_regeneration.py +``` + +Expected: PASS + +**Step 4: Commit** + +```bash +git add docs/generated/complex-support.md docs/math/complex-support.json README.md .github/workflows/oracle-integrity.yml validators/complex_support.py scripts/check_complex_support.py scripts/report_complex_support.py tests/test_complex_support.py tests/test_complex_support_report.py tests/test_math_registry.py tests/test_repo_config.py tests/test_scripts.py +git commit -m "feat: track complex support across the published DB surface" +``` From bc8aa21c69d6b4aeaa6e23a5267145854d9d748c Mon Sep 17 00:00:00 2001 From: Hiroshi Shinaoka Date: Tue, 31 Mar 2026 20:27:35 +0900 Subject: [PATCH 2/4] feat: add matrix_exp oracle cases and document eig complex strategy Materialize matrix_exp cases (8 records across 4 dtypes) by routing through cmi.op_db via a new inventory_kind="cmi_linalg". Update the replay validator to handle the new routing. Add matrix_exp to the math registry and replace the "not yet materialized" note with a proper DB Families section. Document the eig complex oracle strategy covering phase ambiguity, real-to-complex handling, and future considerations. Closes #17 Co-Authored-By: Claude Opus 4.6 (1M context) --- cases/matrix_exp/identity.jsonl | 8 +++++ .../pytorch-upstream-publish-coverage.md | 3 +- docs/math/eig.md | 34 +++++++++++++++++++ docs/math/matrix_exp.md | 8 +++-- docs/math/registry.json | 6 ++++ generators/pytorch_v1.py | 28 +++++++++++++-- tests/test_math_registry.py | 4 +-- validators/replay.py | 4 +-- 8 files changed, 85 insertions(+), 10 deletions(-) create mode 100644 cases/matrix_exp/identity.jsonl diff --git a/cases/matrix_exp/identity.jsonl b/cases/matrix_exp/identity.jsonl new file mode 100644 index 0000000..5d6546f --- /dev/null +++ b/cases/matrix_exp/identity.jsonl @@ -0,0 +1,8 @@ +{"case_id": "matrix_exp_f64_identity_001", "comparison": {"first_order": {"atol": 1.0, "kind": "allclose", "rtol": 0.0001}, "second_order": {"atol": 0.01, "kind": "allclose", "rtol": 0.0001}}, "dtype": "float64", "expected_behavior": "success", "family": "identity", "inputs": {"a": {"data": [-4.826984902407649, -7.582430495695129, -0.6046891126565539, -1.5411692216498662, -1.9719097652168658, -4.146041530864057, -5.4215280659972755, -4.784169829877467, -4.150805063878843, 7.380839390984031, 5.576059452216908, -8.315684908389088, 4.177597026003685, 2.047386382824099, -4.076012721760325, -3.063683029231029, -3.342174545322517, -5.439777184204883, 4.480518929058965, 2.685009210157367, -3.8432258180800494, -3.0355148286483775, -2.146076312776824, -2.6718482427688133, -7.7232222137058715], "dtype": "float64", "order": "row_major", "shape": [5, 5]}}, "observable": {"kind": "identity"}, "op": "matrix_exp", "probes": [{"cotangent": {"value": {"data": [0.03749158605924023, -0.26219801992153097, -0.014738706057789775, -0.06514400708589858, -0.3806125381639458, -0.005511479010301242, -0.1424047905362691, 0.10807689847880969, 0.1340819070106057, -0.17969156067364714, -0.12005126875985751, -0.025470072215545307, 0.12230315381028785, 0.20102178293465053, -0.09382634387476411, 0.45436378816921913, -0.2781584742885612, 0.06817454311935953, 0.0916440284361147, 0.47547025810353694, 0.042151500505492843, -0.27164080295203413, -0.011853571758602353, -0.13710725348529879, -0.03986234991109686], "dtype": "float64", "order": "row_major", "shape": [5, 5]}}, "direction": {"a": {"data": [0.2897559459072697, 0.11813050050423236, 0.010752994487092288, 0.08844143824267293, -0.3953577139982132, -0.05673303740933913, 0.20547366830611524, 0.276837843583669, 0.00154081498253048, -0.2739983842584462, 0.048243247099981734, -0.13121830867052892, 0.31812325233423105, -0.2570781810160961, -0.065924188778598, -0.3081479672618064, 0.06426309546023966, -0.29851992005858313, 0.07302156708519598, -0.04037612264190988, 0.018734891121592358, -0.2779134863846212, 0.08845073857435944, -0.1902216109282594, -0.18096217769148043], "dtype": "float64", "order": "row_major", "shape": [5, 5]}}, "fd_ref": {"hvp": {"a": {"data": [12.978553439944928, -10.023032982599624, 2.138493137887333, 17.57390262234585, 6.157464200574895, -14.408779273322246, 8.720218303245547, -3.4627032964080655, -11.097084141802597, -8.035486143037561, 18.054587613424054, -13.19568135173314, 15.404185022227464, 5.250743271992201, 8.261854706364476, -0.8151901754111295, 5.686344925061556, -34.07995544922302, 33.55605455657628, 5.482256579582898, -11.861321685107265, 7.356127982743483, -8.92198138844809, -0.6792548588327374, -5.979316181037644], "dtype": "float64", "order": "row_major", "shape": [5, 5]}}, "jvp": {"value": {"data": [-32.26480283742171, 65.83327807038532, -10.140184447395008, -115.20644925017707, 31.437171173702744, -27.988803785815374, -20.44100062576798, -85.80249500959167, 144.76144242860056, 11.973120016493587, -166.18805071425365, 277.4748282688571, -97.64245565278017, -399.6806182049697, 152.3504601272933, 341.1401445472401, -383.0643416571987, 398.7003663757165, 236.49185481593412, -271.4416191284343, -33.60573831788138, 23.33778122652091, -53.22049225470371, 22.92281259771126, 24.324128808227673], "dtype": "float64", "order": "row_major", "shape": [5, 5]}}, "method": "central_difference", "stencil_order": 2, "step": 2.323594920282279e-05}, "probe_id": "p0", "pytorch_ref": {"hvp": {"a": {"data": [12.978553439654135, -10.023032979125611, 2.1384931263687026, 17.57390262801297, 6.157464202759279, -14.408779271751222, 8.720218300730005, -3.462703286087295, -11.097084148478238, -8.03548614308561, 18.054587616048867, -13.195681350284309, 15.40418501412267, 5.250743277100978, 8.261854709991987, -0.8151901838020311, 5.686344933227726, -34.07995546353169, 33.55605455406377, 5.482256575246853, -11.861321685477714, 7.3561279825485695, -8.921981384458983, -0.6792548631513118, -5.979316181542599], "dtype": "float64", "order": "row_major", "shape": [5, 5]}}, "jvp": {"value": {"data": [-32.264802874102585, 65.8332781176309, -10.140184502794835, -115.20644926434373, 31.437171206556506, -27.98880370885223, -20.44100074613962, -85.80249491363845, 144.7614425370877, 11.973119943022148, -166.18805090708187, 277.47482851109135, -97.64245593113533, -399.68061830596423, 152.35046029990227, 341.14014462268005, -383.06434165401475, 398.7003665257623, 236.49185459678125, -271.4416191718191, -33.60573830780709, 23.33778120451927, -53.22049224487411, 22.922812626986033, 24.324128797518483], "dtype": "float64", "order": "row_major", "shape": [5, 5]}}, "vjp": {"a": {"data": [5.422514115078143, 14.79270555961115, -75.73867325758332, 62.33713185265172, 20.11013264040867, 4.024304841245779, -9.748833570026513, 58.518625042230184, -50.44011951344014, -7.385460956593755, 29.529655998916912, -2.522270653932809, -37.39438231663237, 45.138722225829625, 30.078470766458683, -64.1718231769342, 62.6709113134741, -132.54751533592614, 13.65306811474737, -18.444896386617316, -12.219029009579463, 6.4049497097548365, 15.035952944434452, -32.042150976062594, -10.489117054436514], "dtype": "float64", "order": "row_major", "shape": [5, 5]}}}}], "provenance": {"fd_policy_version": "v1", "generator": "python-pytorch-v1", "seed": 17, "source_commit": "449b1768410104d3ed79d3bcfe4ba1d65c7f22c0", "source_file": "torch/testing/_internal/common_methods_invocations.py", "source_function": "sample_inputs_matrix_exp", "source_repo": "pytorch", "torch_version": "2.10.0"}, "schema_version": 1} +{"case_id": "matrix_exp_f64_identity_002", "comparison": {"first_order": {"atol": 1.0, "kind": "allclose", "rtol": 0.0001}, "second_order": {"atol": 0.01, "kind": "allclose", "rtol": 0.0001}}, "dtype": "float64", "expected_behavior": "success", "family": "identity", "inputs": {"a": {"data": [-4.826984902407649, -7.582430495695129, -0.6046891126565539, -1.5411692216498662, -1.9719097652168658, -4.146041530864057, -5.4215280659972755, -4.784169829877467, -4.150805063878843, 7.380839390984031, 5.576059452216908, -8.315684908389088, 4.177597026003685, 2.047386382824099, -4.076012721760325, -3.063683029231029, -3.342174545322517, -5.439777184204883, 4.480518929058965, 2.685009210157367, -3.8432258180800494, -3.0355148286483775, -2.146076312776824, -2.6718482427688133, -7.7232222137058715, -8.596239972526284, -0.524154812497627, -3.2519978374491334, 6.846803848769272, -0.6113253421876204, 8.512023091609862, -3.9030392424287763, 5.370812764172914, -1.1798153561819573, -3.1236837738138137, 8.685287845850093, -7.447757767478217, -6.67991226028405, 0.07684976479617145, -3.9762864810578007, -0.19296459711253178, -1.4410484284202578, 8.704956403649721, -6.189088570548082, 5.247816827502734, -5.054033686620959, 4.098306048998678, 5.669103499758022, -6.102503890250265, -7.630780794546761, -6.127846976735512, 0.5886329087259614, -8.016679309730627, 6.350501338142502, 0.37123771033595565, -8.501911864528664, 8.881640064966652, 5.091308063884989, 2.2682829542038476, 7.965859477001853, -5.508718606208063, 5.929521921772405, -6.550108688041963, 4.3697425449165594, 8.750419690856972, -3.0004209767941776, 8.567080406475506, -6.37727772642512, 6.023749307539076, -0.14263441434025048, -0.08620258275208337, 5.361116377951722, 7.261913172012517, -1.8770140603173509, -1.0866502343265727, -6.683097295281247, 0.2021654305995635, -1.36447028146176, 3.870594897435672, 3.8557350299919806, 4.629809887990001, 3.26047253875904, -0.5460588607984951, 0.15655662337879184, 6.894402913350771, 6.469925347984147, 7.881469215756416, 7.2418997658575215, 6.666640236740337, 2.8553673151215015, -0.5571856086612246, -7.147824109565546, 5.459685046361393, 0.46995756502570907, 4.960804054305201, -4.64425751027945, 0.477847484149605, 4.064960411352818, 8.5109468441137, 8.563048232451571, -8.012767077428752, -5.035128194061096, 8.711754089536855, -3.7047515579502805, -8.40220690684875, -2.2081080737555254, 3.9072623108906424, -3.7103554852154486, 1.9178346173798106, -4.345763640591077, -8.121701800994094, 0.5594417733989336, 1.7686045670680375, 8.720165669683919, -2.994344215122795, -2.572983461238471, 3.996264006153404, 7.950957679746697, 6.809653481934188, 8.454201183688609, -5.462610544573957, -2.162002455421761, 4.094330512933896, -7.7581383157545325, -4.959005925438627], "dtype": "float64", "order": "row_major", "shape": [5, 5, 5]}}, "observable": {"kind": "identity"}, "op": "matrix_exp", "probes": [{"cotangent": {"value": {"data": [-0.07232432668455255, 0.0754333297307769, 0.10303619812193575, 0.09976501807748045, 0.08894208405401823, -0.04192583761246802, -0.10959111477779536, 0.03391612699898509, 0.0543865366072637, -0.06717885209713342, -0.06457400725942479, 0.07955164604242575, 0.12508636118686225, -0.006829800581029771, 0.021541598397551296, 0.11013789404649528, 0.15537196110632456, -0.04990675191153632, 0.16432160471745788, 0.053511939240994726, -0.1775997007340156, 0.025098172226793086, 0.04057724273261239, 0.1784071954838335, 0.010722185787725802, -0.014173566736384527, 0.038398116978659365, 0.1457802201742064, 0.047681943523300734, -0.03920950933455012, -0.06955073341823141, 0.03212551791372982, 0.031216292340102644, 0.020827161133653297, 0.029636459519470214, -0.08099061190202496, -0.1775959710174149, -0.010491931834667945, 0.2212299883175103, 0.008192655011715315, 0.02787421575698276, -0.016853190009994055, -0.033287000752802165, 0.001383960260601717, 0.020410063938864596, 0.0043491737519329495, -0.008473145999889777, -0.07092099979766323, 0.11711490009833694, 0.02283549629684743, 0.039743719775622896, 0.04967293808485848, 0.13851811347136334, 0.0232306756861612, 0.018219674152057302, 0.03924743071753622, 0.04860225185512067, -0.049991324981619886, 0.13824981392011507, -0.10945811646302052, -0.05827202170975786, 0.23807198155429693, -0.07363162020567623, -0.0772359098242121, 0.06475093753458659, 0.017241123287419783, 0.010844630513788731, -0.13020228788650828, 0.1180910654136767, -0.05967975182870797, 0.0861513311050679, -0.11523810476932146, 0.10961681309608713, -0.22388244016461573, 0.03904609845187731, 0.1736945568156921, -0.08972870231439946, 0.0842909599204131, 0.08402618877119829, -0.036816048877772296, -0.0035045699515795985, 0.06667611930901225, 0.0269680352838897, 0.07814833730835496, 0.018229722424984437, 0.120130339274635, -0.10088113627138226, 0.005192434593282756, 0.01179013066722322, 2.247514033922583e-05, -0.04172074744332103, -0.03261268402183222, 0.08059422529321517, -0.05332759928473539, -0.08584538016213335, -0.005291950555466408, 0.027068326303409745, 0.11914397084112058, 0.08336846602567412, -0.005044402475954505, 0.043070092927485634, 0.004849971616702539, 0.07139738702085133, 0.034187702371157, -0.04909689089758047, -0.007928915192688103, -0.010964143241416403, 0.2444908941285295, 0.1426418387003382, 0.04780782682231287, 0.005478514909875523, 0.21267620724673947, 0.12479920876760375, -0.03076273066231769, 0.12156384262768377, -0.11324893903327385, 0.0890606872441571, 0.07210450185230217, 0.10268987883415497, 0.1532147645820999, -0.026067368771707074, -0.10178449746457045, -0.06034640664017117, 0.00398133989828399, -0.025399047039532337], "dtype": "float64", "order": "row_major", "shape": [5, 5, 5]}}, "direction": {"a": {"data": [-0.03295713549354109, 0.05652071080325509, 0.08787920707787512, -0.027528113276212882, -0.036012022847698416, -0.0580862458092365, -0.05434966572452121, -0.004803309980408778, 0.2444230012149746, -0.010588701525025022, -0.02688190922529404, 0.09305610601974931, -0.08412422446654573, -0.049905340565114906, -0.02279439788883356, -0.06479196322260976, -0.08802473416305048, -0.12930306725323057, 0.15077733441300586, 0.06943191665848719, 0.02148119226857402, 0.05739515204013026, -0.037060092825650995, 0.05739710609892657, 0.19082000301789773, 0.012674056772835382, 0.07994281397282742, 0.2028568409067504, -0.19968823304837036, -0.13515058113195533, -0.177128233363998, -0.006933404929653769, -0.0037348563654430776, 0.06563017412787477, 0.1254233971856089, 0.10970310744112984, 0.15295435250521977, 0.10042314394879645, -0.003121209004466388, 0.1153934707600631, 0.0843784947343007, -0.09354749319307824, -0.01419628612316597, 0.049862511636746605, -0.07243877781078074, 0.027824760754579635, -0.004026839824636022, 0.02210153405612477, -0.03618196425671232, -0.06601547740428033, 0.1336297690943561, 0.03799201364328763, -0.03567813107838834, -0.03448040187524411, 0.05308178623437534, 0.12119928091590343, 0.004376546983611136, -0.0657536423583619, 0.033313633199310494, 0.04820949095090037, 0.006239575057652329, -0.011414672609675217, 0.08768723251506509, -0.010517491799502055, 0.06452055195201707, -0.06429207592786852, -0.006441974866054176, 0.018910413633762196, 0.10483856415700565, 0.07055425482942995, 0.1625585099711579, 0.09986210094423364, -0.05000295464182405, -0.04103780691774508, -0.05761114172575383, -0.10697211691553785, -0.10699727150422972, -0.007464978546943462, 0.19785709989364908, -0.016517020496397833, 0.12810093643262155, 0.024539766385105393, -0.2400151107243466, -0.1410456538305126, -0.027049302156521734, 0.08162767744160208, -0.011472165895758418, -0.17110077366388066, -0.0014663248767418297, 0.07079765311425913, 0.040749959013884084, 0.1559591134070347, 0.10416780652745662, 0.022045941061417575, 0.02468126588841372, 0.05451007732647208, 0.12695209775437297, -0.03617524028886253, -0.07452392006783469, 0.05480355416496389, 0.03541828176095958, -0.06040082165406976, 0.2035804183461476, 0.05522253529279824, -0.021479684287537052, -0.019523590985835556, 0.11251167216314693, -0.007172523078974828, 0.011158134498382134, -0.09721335713698094, 0.06748838047491684, 0.038608963388484435, -0.11250001064362802, -0.08095993271540299, 0.14759053874459727, -0.026451012549839132, 0.08975354431300187, -0.10652497225552283, -0.06804982008524313, 0.048358999653460674, 0.06973908765445462, 0.08512561449365323, -0.04542958399748188, -0.04764866634170121, -0.007676078717390373], "dtype": "float64", "order": "row_major", "shape": [5, 5, 5]}}, "fd_ref": {"hvp": {"a": {"data": [-1.195559557757879, 1.4006487078812266, -2.1861941975933163, 0.5070681629171877, 0.30102441231080684, 1.7082628822352686, -1.457796753377888, 0.9750172299802771, 1.2364433660518976, -0.4926584372165621, -1.6054149914174365, 2.2260610620284376, -4.178702274997969, 2.486432645054879, 0.27491130499390587, -0.9863538133025845, 0.09761373832247235, 4.35571445191563, -6.629251359999315, 0.775601311358574, 1.5541648029813042, -1.7441343638526245, 3.049120355300313, -1.8268996256631445, -0.297384035148143, -0.0003383628717112579, -0.0004510354372516839, -0.0007580761405397931, -0.0009622678667957631, 0.00023358796651354124, -7.08550946432329e-05, -0.0006108597946076009, 0.0003158097172595648, 0.00022256859862958666, -9.576995109754338e-05, -9.377211951198123e-05, 0.0001997647665689457, -0.0004956276066732437, -0.0004940002244605628, 0.00012781557543968297, -0.00016796914601118068, 0.00014066339707336238, -0.0007153345258391576, -0.0007814310971049127, 0.00022082192429936825, 8.764885552361596e-05, 0.00041110704236492073, -4.9232002942011955e-05, 4.919356228158986e-05, -7.866953801713985e-07, -434.781118601833, -3705.3924723728815, -731.7036206699678, -1586.7576412420171, -1295.4432026343666, 2082.1527542435397, 7272.526041208704, 448.16983642982655, 4714.902795278843, 1248.9843143397075, 923.7337227363927, 8080.751834365513, 3039.787724283007, 4719.362348914762, 3184.1591164014917, 310.25201687310096, -697.4415708540276, -906.4129315505695, -184.2225944137438, -796.4565016002641, 1363.0878771750008, 5358.275822650807, 650.8220200003728, 3378.5605943078635, 1232.0626261892703, 996.3441155525676, -51.22697130975341, 3083.6627949487433, 2859.932072283904, 3145.79728993306, 3458.243147413932, 2729.46749340357, 14133.73553384446, 9440.38491106237, 12756.233658295892, 3309.2240323511564, -15506.581198994869, -7178.77954004185, 12172.725362599795, 960.0392334744079, 4871.002198452641, -12936.774249429058, 975.7865032636352, 15939.85208892355, 6952.821931448304, 10042.635926561818, -8267.480002120508, 22416.073070241055, 30129.959047670214, 26798.773146298277, 228.1163102932666, -1.5164971940067054, 761.8768882776852, 867.9014574772416, -247.33312200143976, -260.78330261904813, 20.192380195139545, -971.8115145121852, -1105.5933022709723, 287.2648213823147, -231.18932820464636, 7.101358902371299, -638.6241500280283, -724.9136894700558, 241.1640925140682, -447.28689961474447, 23.380467270152938, -1402.5549600999616, -1593.57560130314, 476.7236515403478, -208.9932247711422, -4.398174405812678, -718.9856161399073, -820.1238226046195, 228.54917678552303], "dtype": "float64", "order": "row_major", "shape": [5, 5, 5]}}, "jvp": {"value": {"data": [-15.611456748345597, 2.4505006227797366, -26.114343780433757, 31.872837024746715, 13.152593496535141, 25.95659287495133, -15.263672236461842, 36.01903215590919, -26.454960530070487, -23.36404977284904, -21.08907998550798, -12.492872714507055, -39.0096346065428, 82.80043230162813, 23.578346183448225, -1.1532885080355288, 52.2542059089759, 27.798461939222214, -125.18091051286206, 2.9597812125527185, 9.659212476724468, -12.065574362923897, 10.156301370871567, 7.903783176854517, -8.96239735056656, -0.004181952629077787, 0.0009895149621144438, -0.0015060537833368275, -0.004167209484502021, 0.000466780600389957, -0.009816330738794187, -0.0016290011974965893, -0.0009492596693974518, -0.006083869687372965, 0.0041711332816745045, -0.0025126296486848365, 0.006038260792367902, -0.004888920807412318, -0.0075067851392634265, -0.0038666310343197334, -0.0056002090066614146, 0.005060411639276131, -0.004688464646737239, -0.009289986535770447, -0.0025113648980167147, 0.0011047040969721122, -0.0008626987484185521, 0.0003786733961339389, 0.0015563702600437457, 0.00028141179328838807, -6125.778201392354, 8628.22176216143, 6727.182387747347, -543.1980899361646, 7129.181145412766, -62765.8453923465, 285745.32784673315, 108031.7485130372, 50394.60086335496, 178796.35456007486, -47893.61829790133, 178903.3849185937, 74683.30125120535, 27351.424984163874, 115463.1955920201, -54710.005137796485, 184745.8122137833, 81423.08130843236, 25682.25377241567, 121381.04090089687, -35536.699741500765, 145538.61234011827, 57948.68893543506, 23923.428609081588, 92527.82855189659, 9044.903226138089, 24984.623850071694, 68302.9120706062, 80765.64354684124, 112080.89883416612, 12242.819672611568, 50509.54360831768, -29586.780116072645, 12590.732452612538, 87639.93612157447, 39830.75376418791, 133386.74035049602, 129997.01230771096, 220297.76891841795, 403912.5961296018, 26878.6551485662, 70694.77859737314, 228944.46656993413, 260592.4311976658, 346699.11266443436, 43481.29451431945, 128778.18297487486, 264965.9324578151, 338021.1627578534, 505525.6848760037, 741.8273402969544, -868.7600770831561, -530.1120414186939, -1192.4904771441693, -747.5417694109256, -200.38535235247508, 307.541775888821, 107.82995784652205, 332.3653218908186, 197.77089940769304, 1557.1932223578399, -2145.5034477857325, -918.0237258353047, -2490.10074992507, -1568.4739307974085, 1731.434675932681, -2403.3451976539113, -1007.6663543505605, -2764.6654912814975, -1744.6899635643165, -713.1464760111744, 857.3037914009142, 493.8676693962062, 1141.9077272045436, 719.6840451460412], "dtype": "float64", "order": "row_major", "shape": [5, 5, 5]}}, "method": "central_difference", "stencil_order": 2, "step": 5.9986792520837365e-05}, "probe_id": "p0", "pytorch_ref": {"hvp": {"a": {"data": [-1.195559558501926, 1.4006487085583266, -2.1861941991462825, 0.5070681624681964, 0.30102441255560464, 1.708262883203045, -1.4577967543297854, 0.9750172320126276, 1.2364433661156207, -0.49265843751201593, -1.6054149921412886, 2.2260610628910067, -4.1787022773523015, 2.4864326446794514, 0.27491130522941215, -0.9863538137379961, 0.09761373876197436, 4.355714451437846, -6.629251359421084, 0.7756013114136824, 1.5541648035579645, -1.7441343646086709, 3.049120356643442, -1.8268996247300346, -0.29738403542774783, -0.0003383628718729623, -0.00045103543757082253, -0.0007580761406752644, -0.0009622678670646631, 0.00023358796656668983, -7.085509459707068e-05, -0.0006108597948450237, 0.0003158097175723238, 0.00022256859899171072, -9.576995114384913e-05, -9.377211966647257e-05, 0.00019976476647695998, -0.0004956276070053847, -0.0004940002249125189, 0.00012781557552013557, -0.00016796914615777885, 0.00014066339696084773, -0.0007153345261132966, -0.0007814310975065628, 0.00022082192437667905, 8.764885551606249e-05, 0.00041110704241503445, -4.9232002985723247e-05, 4.9193562227064575e-05, -7.866953865515233e-07, -434.7811186564363, -3705.3924729265937, -731.7036209033338, -1586.7576419896002, -1295.4432035512566, 2082.152754487129, 7272.526042023806, 448.1698370417792, 4714.902796354781, 1248.9843156620109, 923.7337226430931, 8080.751834223536, 3039.787724196681, 4719.362348877754, 3184.1591163825797, 310.2520168963852, -697.4415705051705, -906.4129313310237, -184.22259405851113, -796.4565012916439, 1363.087876966918, 5358.275823410175, 650.8220205187874, 3378.560594826309, 1232.0626274704703, 996.3441157546453, -51.22697087119491, 3083.6627957991027, 2859.932072705067, 3145.79729069831, 3458.243148051492, 2729.467494797513, 14133.73553719522, 9440.384912506572, 12756.233660837983, 3309.224034158087, -15506.581195667946, -7178.7795319579855, 12172.72536599436, 960.0392403144183, 4871.002200862411, -12936.77424620785, 975.7865130833557, 15939.85209314748, 6952.821938418496, 10042.635929565313, -8267.47999733453, 22416.07307982322, 30129.959052235074, 26798.7731533457, 228.11631033887542, -1.5164971883552534, 761.8768883486066, 867.9014575576058, -247.33312205416698, -260.7833026909878, 20.19238018075751, -971.8115146648765, -1105.5933024305607, 287.26482147117855, -231.18932822668975, 7.10135890078396, -638.6241500310044, -724.9136894714352, 241.1640925377331, -447.28689969178265, 23.380467259469974, -1402.5549601928924, -1593.5756014220274, 476.7236516268633, -208.99322481701546, -4.398174411390977, -718.9856162135256, -820.1238226880604, 228.54917683591785], "dtype": "float64", "order": "row_major", "shape": [5, 5, 5]}}, "jvp": {"value": {"data": [-15.611456744273378, 2.4505006162805363, -26.11434377932909, 31.872837035582915, 13.152593491650729, 25.956592877334813, -15.263672233343117, 36.019032165645335, -26.45496055136678, -23.36404977211184, -21.08907998213899, -12.49287273282831, -39.009634619349725, 82.80043235084945, 23.578346175394362, -1.1532885394194805, 52.2542059410957, 27.79846190753245, -125.18091051801619, 2.9597812386095823, 9.659212478257565, -12.065574362756983, 10.156301374038383, 7.9037831710871, -8.962397351082526, -0.004181952630829952, 0.0009895149620700939, -0.0015060537844655045, -0.004167209486120124, 0.0004667806008019484, -0.009816330741567679, -0.0016290011996113276, -0.000949259670280796, -0.006083869688118701, 0.004171133283286862, -0.002512629649656187, 0.006038260795352971, -0.004888920809785529, -0.007506785142488116, -0.0038666310351033713, -0.00560020900873038, 0.005060411642239491, -0.004688464649870346, -0.009289986539903712, -0.0025113648985069766, 0.0011047040982399398, -0.0008626987475688523, 0.0003786733962771209, 0.0015563702608795932, 0.00028141179266778776, -6125.778201996724, 8628.221763700461, 6727.182388792635, -543.198089913295, 7129.181146710568, -62765.84522798273, 285745.32750254095, 108031.748314327, 50394.600855827084, 178796.35428772744, -47893.618216175484, 178903.3847489661, 74683.30114845956, 27351.42498049934, 115463.1954564788, -54710.00505331899, 184745.81203600703, 81423.08120332811, 25682.253768591, 121381.04076590111, -35536.69965747008, 145538.61216312528, 57948.688822726785, 23923.42860509463, 92527.82841076012, 9044.903226857441, 24984.623853899422, 68302.9120799129, 80765.64355320128, 112080.89884430757, 12242.819675763732, 50509.54362124868, -29586.780083698646, 12590.732468985647, 87639.93615954327, 39830.75376886645, 133386.74037491312, 129997.01236024992, 220297.76893587818, 403912.5961771924, 26878.65514954939, 70694.77860587738, 228944.46657702385, 260592.43119825807, 346699.1126819294, 43481.29451752773, 128778.18299473205, 264965.93249806017, 338021.1627682259, 505525.6849041932, 741.8273403291073, -868.760077251823, -530.1120413509035, -1192.490477177839, -747.5417694568578, -200.38535233602923, 307.5417758771787, 107.82995782375332, 332.3653218539364, 197.77089939609658, 1557.193222229415, -2145.5034478537464, -918.0237255988454, -2490.1007496766288, -1568.473930688466, 1731.434675796016, -2403.3451977170375, -1007.6663540839378, -2764.6654910006196, -1744.6899634502415, -713.1464760166718, 857.3037915404623, 493.8676693179359, 1141.9077271899018, 719.6840451633673], "dtype": "float64", "order": "row_major", "shape": [5, 5, 5]}}, "vjp": {"a": {"data": [6.79892478774711, -7.195812428976788, -28.13135701801443, 46.32041829569242, -4.856212591021592, -2.5474187801222525, 10.926442706494864, 31.800431598978054, -48.67876342237622, 4.994824661328921, 11.719835460139286, -12.133114953095735, -29.52877836953997, 70.43333371207326, -7.86983803642188, -17.555967026144316, 5.512343140483626, -48.80950510764907, 7.6731863509195275, 2.9706631175218003, -2.330609085060377, 7.941437566806233, 30.578796109792044, -57.37893953981295, 5.157035024395262, -0.0019956503293740236, -0.010550440457272643, 0.0033021475252107725, 0.000873742480677905, -0.0005469869860335882, 0.0036325365481003568, 0.0016906558868032772, 0.010075597982004074, 0.012881717314373256, -0.0042188448467609125, -0.0038667844856096296, -0.008146830621006018, -0.004736289124611381, -0.00830378991977561, 0.002605598671152087, -0.0028939670248970414, -0.009178546759265337, -0.0018739811617030583, -0.004968449331608681, 0.001195867512676896, -0.0010878355462929068, -0.0006496644129137771, -0.0029424881044180645, -0.0038987817264571276, 0.0013161464044079737, -37299.78914970785, 102097.24432932511, 90331.71219081857, 9803.250005527892, 88735.97026285983, 69031.97231793366, -147147.74884877002, -145710.4857165578, 3118.080818024755, -145394.76205097788, 41730.60509529222, 36115.21210967303, -23873.331449399928, 65476.96337170995, -31569.051160800376, 2344.63046622923, -50183.23172002604, -28141.1402898643, -22864.641064169737, -25282.530715196666, 51447.9822002074, -72995.2167401139, -89779.08172887686, 20961.35485745553, -91853.58006274367, 12356.581814238036, 24512.818099182943, 69536.7353226178, 30570.065143882624, 53288.133777433635, 25342.446317690355, 55431.220848571174, 148089.12212572052, 63454.20162141857, 116162.70763873283, 141983.34246101734, 228144.18377663702, 742233.2778313154, 343414.42803670553, 540994.5159287048, 140908.46139177075, 229010.81519689187, 739368.2829153583, 341195.5000290795, 540355.1730315901, 175701.320013285, 317438.25233550946, 955760.5736776264, 430119.8156575447, 716258.0813484215, -995.6208775699768, -39.21642321925328, -2247.871906827666, -2537.8937731009873, 1010.2721736680239, 1678.730335708211, 131.81299120668615, 3982.8408948481906, 4510.533648761048, -1712.1840427198304, 586.8132430080492, -48.39533065072408, 1110.8440253963072, 1234.3084390838426, -574.2387038089006, 1826.4774422847897, 29.098451288442767, 3994.7051488597826, 4495.246286849395, -1832.95555403416, 929.7757763182449, 47.143130299234116, 2131.560535510852, 2410.23977865167, -948.3336493578497], "dtype": "float64", "order": "row_major", "shape": [5, 5, 5]}}}}], "provenance": {"fd_policy_version": "v1", "generator": "python-pytorch-v1", "seed": 18, "source_commit": "449b1768410104d3ed79d3bcfe4ba1d65c7f22c0", "source_file": "torch/testing/_internal/common_methods_invocations.py", "source_function": "sample_inputs_matrix_exp", "source_repo": "pytorch", "torch_version": "2.10.0"}, "schema_version": 1} +{"case_id": "matrix_exp_c128_identity_001", "comparison": {"first_order": {"atol": 10.0, "kind": "allclose", "rtol": 0.0001}, "second_order": {"atol": 1.0, "kind": "allclose", "rtol": 0.001}}, "dtype": "complex128", "expected_behavior": "success", "family": "identity", "inputs": {"a": {"data": [[-4.826984902407649, -7.582430495695129], [-0.6046891126565539, -1.5411692216498662], [-1.9719097652168658, -4.146041530864057], [-5.4215280659972755, -4.784169829877467], [-4.150805063878843, 7.380839390984031], [5.576059452216908, -8.315684908389088], [4.177597026003685, 2.047386382824099], [-4.076012721760325, -3.063683029231029], [-3.342174545322517, -5.439777184204883], [4.480518929058965, 2.685009210157367], [-3.8432258180800494, -3.0355148286483775], [-2.146076312776824, -2.6718482427688133], [-7.7232222137058715, -8.596239972526284], [-0.524154812497627, -3.2519978374491334], [6.846803848769272, -0.6113253421876204], [8.512023091609862, -3.9030392424287763], [5.370812764172914, -1.1798153561819573], [-3.1236837738138137, 8.685287845850093], [-7.447757767478217, -6.67991226028405], [0.07684976479617145, -3.9762864810578007], [-0.19296459711253178, -1.4410484284202578], [8.704956403649721, -6.189088570548082], [5.247816827502734, -5.054033686620959], [4.098306048998678, 5.669103499758022], [-6.102503890250265, -7.630780794546761]], "dtype": "complex128", "order": "row_major", "shape": [5, 5]}}, "observable": {"kind": "identity"}, "op": "matrix_exp", "probes": [{"cotangent": {"value": {"data": [[-0.11753722537078294, -0.07852618664658965], [-0.016660112511603927, 0.07999915688341805], [0.13148943955213666, -0.06137232090578678], [0.2972018205538369, -0.18194496813696226], [0.04459333876966812, 0.05994485652971331], [0.31100767712356836, 0.027571525318480755], [-0.17768172393171325, -0.007753485639595309], [-0.08968259885136619, -0.02607417948703214], [0.16009763003065808, 0.11626851097770495], [-0.09136879343657907, -0.04147795827070133], [0.04933776016028434, 0.0809388220642407], [0.17793998855428814, -0.18473326144993762], [0.1390617593806392, 0.0012373813402248445], [0.03774701396116793, -0.2150578873428805], [0.1464069420421213, 0.11132251798837214], [-0.12217581306061519, 0.13252098660430292], [-0.12305332584192075, 0.19822355027856453], [-0.07545589806734773, 0.11254336096721809], [0.037303214501424244, -0.1578863514091379], [0.018085944655513397, 0.11576270434818006], [-0.05964488639187716, -0.05246928044928915], [-0.3169735511741073, -0.019047600083066538], [0.19860524392306328, -0.10420450136976456], [-0.2685194311830636, 0.08096404227112103], [-0.13559576264494702, -0.25054506152563993]], "dtype": "complex128", "order": "row_major", "shape": [5, 5]}}, "direction": {"a": {"data": [[0.16889942624163376, 0.06885854816310531], [0.0062679459210495956, 0.05155272354601773], [-0.23045494664627963, -0.03306982169901389], [0.11977108727139989, 0.16136943384146005], [0.0008981446978734174, -0.20208982788110583], [-0.10878053460994834, -0.009332117058425115], [-0.08829242664363632, -0.08507847247472002], [-0.17713587788301688, 0.23218458987233936], [0.4115581159968341, -0.0506109463963323], [0.10993074962846325, 0.1268814992381575], [-0.23761588477423234, -0.19907487290692719], [0.05957196694512519, -0.07722728627569621], [0.1511563906517827, -0.22799404766117465], [0.041077215155125614, 0.14261554177857586], [-0.06128696021345076, -0.3412611043289706], [0.037229163947317546, -0.08574816150789193], [0.03745911034805619, 0.03889775174507771], [-0.14432064669676795, -0.16533951170820332], [-0.15429898583571455, -0.1725723626605022], [-0.07650164684498123, -0.08868513599290274], [0.042613600621509186, 0.017043840222180702], [-0.055428532696050824, 0.1678845943968219], [0.07637835172513949, 0.01746422427218361], [0.0869358085276681, -0.12272849150710842], [0.06180075450046607, 0.022203105791807877]], "dtype": "complex128", "order": "row_major", "shape": [5, 5]}}, "fd_ref": {"hvp": {"a": {"data": [[4.6497804904587134, 12.73425641830263], [-41.7880708130941, 23.71686913213387], [3.4916601285281517, -21.26036262704997], [-18.824381922237553, -2.3620653527689734], [-21.627102852011244, -17.67001385344975], [23.5815909199889, -0.4986210438810776], [15.27487865145725, 84.93040961156164], [-33.16154766436992, -18.149152108889577], [-14.800577142828972, 28.91165753383551], [-42.26642300490601, 25.568773646064137], [-9.415882909568227, 1.0175216419080861], [-9.963431713471296, -34.377732255133914], [14.265117391255963, 6.218809510577356], [5.341303304607002, -11.969473295830936], [16.564317411026675, -11.751973521890127], [-0.6316524961333214, -3.529982707613], [14.393510050813543, -6.054031792784547], [-1.9862417287265381, 5.816985189334522], [4.411227744171585, 2.2663303089972335], [5.528262716386655, 6.744029333783122], [12.69986891552994, -2.856789843830629], [18.215982371367943, 42.5329168714825], [-19.62294935972786, -5.9456085167361685], [-4.718486944631183, 17.73309431867431], [-19.04851596770677, 19.023420747051993]], "dtype": "complex128", "order": "row_major", "shape": [5, 5]}}, "jvp": {"value": {"data": [[-12.867738015808076, 59.233483323039756], [-113.6344402858692, 3.189732651708734], [43.87852292710815, 3.5608216339124947], [11.14843375058386, -21.339634803999466], [-55.27094545930228, -15.355324690473589], [166.5982349652763, 88.19595631599189], [-86.34979582860635, 348.94253419778556], [51.966978001286144, -133.3137707629846], [-57.3040983809226, -56.168781472605794], [-90.18826121210019, 152.57321868817587], [-28.145984412772318, -90.81911164657214], [153.9013275422361, -92.74544272519361], [-64.02626401443061, 28.80973114956293], [0.7475409023972569, 38.94770087040245], [87.15289612207562, -22.032410362600576], [91.0238725901738, -20.256133662338343], [77.92908649026396, 155.38315410423405], [-24.27645335503302, -62.39534520164258], [-36.46589642930059, -0.7126854388881388], [13.738231261191032, 87.24948601543717], [84.42894081198983, -71.69031209148241], [170.28295595768932, 120.71587898950206], [-61.72949127949578, -54.32816609802181], [-40.804237498495745, 20.244885183113272], [62.904341604268176, 83.1356666949867]], "dtype": "complex128", "order": "row_major", "shape": [5, 5]}}, "method": "central_difference", "stencil_order": 2, "step": 3.6119439304087967e-06}, "probe_id": "p0", "pytorch_ref": {"hvp": {"a": {"data": [[4.649780509567715, 12.734256443758502], [-41.78807088362894, 23.716869247183343], [3.491660115676873, -21.260362693777214], [-18.824381976998335, -2.362065337764146], [-21.627102932091187, -17.670013879049154], [23.58159099725949, -0.4986210655462857], [15.274878858085167, 84.93040985883701], [-33.16154782229019, -18.14915213509933], [-14.800577157975928, 28.911657673647387], [-42.266423127768576, 25.568773809227864], [-9.415882940326174, 1.0175216658576896], [-9.963431848555599, -34.377732333760044], [14.265117465616054, 6.218809493585356], [5.341303285963251, -11.969473358974112], [16.56431743489156, -11.751973618298088], [-0.6316525096165782, -3.5299827138129407], [14.393510051707398, -6.054031862161269], [-1.986241707670544, 5.8169852146448715], [4.411227763855437, 2.2663302898081277], [5.528262759261088, 6.744029324754982], [12.699868950684799, -2.8567898578436406], [18.215982480108593, 42.53291697868629], [-19.622949435363854, -5.945608520404963], [-4.718486943572726, 17.73309438515412], [-19.048516015919745, 19.02342083360146]], "dtype": "complex128", "order": "row_major", "shape": [5, 5]}}, "jvp": {"value": {"data": [[-12.867738120738338, 59.23348339184073], [-113.63444052276625, 3.1897324599170336], [43.878523008416465, 3.5608217447409656], [11.148433832275561, -21.339634795273675], [-55.27094553717551, -15.355324788542825], [166.59823506558874, 88.1959567820849], [-86.34979678300999, 348.94253468966144], [51.96697847787451, -133.313770856801], [-57.304098242282514, -56.16878174075051], [-90.1882616685911, 152.57321881737127], [-28.145984313332633, -90.81911188714002], [153.90132812500394, -92.74544263917366], [-64.02626425618027, 28.809731038553302], [0.74754075607891, 38.94770094371673], [87.15289636103304, -22.032410285113322], [91.02387277633143, -20.256133561282564], [77.92908637736898, 155.3831545860228], [-24.27645324970106, -62.39534539774444], [-36.46589648080636, -0.7126855677028074], [13.738231173159459, 87.24948620762588], [84.42894107212858, -71.69031209386591], [170.2829561201921, 120.71587959671552], [-61.72949127687167, -54.32816637760816], [-40.80423762882854, 20.24488507070521], [62.90434163009637, 83.13566696702473]], "dtype": "complex128", "order": "row_major", "shape": [5, 5]}}, "vjp": {"a": {"data": [[29.268468102501913, 30.328589979680107], [-68.2025748286663, 121.93090158562912], [-20.19028098209598, -61.856092585880916], [-62.50793001315049, 19.338794791453388], [-88.54031689368878, -14.037259672106277], [78.01530798524823, -38.98392981805175], [204.08217916298815, 208.601840028385], [-134.10815164287027, 2.1535110548728262], [-1.3064721374342305, 136.01738476236733], [-80.56633101902682, 169.3840145891835], [-26.84635881917335, 28.50622726040819], [-117.9101967274083, -56.09878959965807], [56.12767966370429, -21.70872118860334], [-20.665055853821684, -57.449501039125465], [7.621985073770394, -83.58988585266682], [-19.17637592541774, -8.445097866196257], [6.13816065666561, -71.72240268624746], [21.130221734764124, 24.20947527942675], [25.66869009962376, -20.36293166299817], [44.73574261526391, -12.147071383316618], [35.92455918696649, -22.121940908872023], [106.79716896328135, 90.21212608708024], [-64.44288553111384, 6.560966173135041], [5.328026320780434, 65.9221451312496], [-30.695314771902712, 84.50004181112271]], "dtype": "complex128", "order": "row_major", "shape": [5, 5]}}}}], "provenance": {"fd_policy_version": "v1", "generator": "python-pytorch-v1", "seed": 17, "source_commit": "449b1768410104d3ed79d3bcfe4ba1d65c7f22c0", "source_file": "torch/testing/_internal/common_methods_invocations.py", "source_function": "sample_inputs_matrix_exp", "source_repo": "pytorch", "torch_version": "2.10.0"}, "schema_version": 1} +{"case_id": "matrix_exp_c128_identity_002", "comparison": {"first_order": {"atol": 10.0, "kind": "allclose", "rtol": 0.0001}, "second_order": {"atol": 1.0, "kind": "allclose", "rtol": 0.001}}, "dtype": "complex128", "expected_behavior": "success", "family": "identity", "inputs": {"a": {"data": [[-4.826984902407649, -7.582430495695129], [-0.6046891126565539, -1.5411692216498662], [-1.9719097652168658, -4.146041530864057], [-5.4215280659972755, -4.784169829877467], [-4.150805063878843, 7.380839390984031], [5.576059452216908, -8.315684908389088], [4.177597026003685, 2.047386382824099], [-4.076012721760325, -3.063683029231029], [-3.342174545322517, -5.439777184204883], [4.480518929058965, 2.685009210157367], [-3.8432258180800494, -3.0355148286483775], [-2.146076312776824, -2.6718482427688133], [-7.7232222137058715, -8.596239972526284], [-0.524154812497627, -3.2519978374491334], [6.846803848769272, -0.6113253421876204], [8.512023091609862, -3.9030392424287763], [5.370812764172914, -1.1798153561819573], [-3.1236837738138137, 8.685287845850093], [-7.447757767478217, -6.67991226028405], [0.07684976479617145, -3.9762864810578007], [-0.19296459711253178, -1.4410484284202578], [8.704956403649721, -6.189088570548082], [5.247816827502734, -5.054033686620959], [4.098306048998678, 5.669103499758022], [-6.102503890250265, -7.630780794546761], [-6.127846976735512, 0.5886329087259614], [-8.016679309730627, 6.350501338142502], [0.37123771033595565, -8.501911864528664], [8.881640064966652, 5.091308063884989], [2.2682829542038476, 7.965859477001853], [-5.508718606208063, 5.929521921772405], [-6.550108688041963, 4.3697425449165594], [8.750419690856972, -3.0004209767941776], [8.567080406475506, -6.37727772642512], [6.023749307539076, -0.14263441434025048], [-0.08620258275208337, 5.361116377951722], [7.261913172012517, -1.8770140603173509], [-1.0866502343265727, -6.683097295281247], [0.2021654305995635, -1.36447028146176], [3.870594897435672, 3.8557350299919806], [4.629809887990001, 3.26047253875904], [-0.5460588607984951, 0.15655662337879184], [6.894402913350771, 6.469925347984147], [7.881469215756416, 7.2418997658575215], [6.666640236740337, 2.8553673151215015], [-0.5571856086612246, -7.147824109565546], [5.459685046361393, 0.46995756502570907], [4.960804054305201, -4.64425751027945], [0.477847484149605, 4.064960411352818], [8.5109468441137, 8.563048232451571], [-8.012767077428752, -5.035128194061096], [8.711754089536855, -3.7047515579502805], [-8.40220690684875, -2.2081080737555254], [3.9072623108906424, -3.7103554852154486], [1.9178346173798106, -4.345763640591077], [-8.121701800994094, 0.5594417733989336], [1.7686045670680375, 8.720165669683919], [-2.994344215122795, -2.572983461238471], [3.996264006153404, 7.950957679746697], [6.809653481934188, 8.454201183688609], [-5.462610544573957, -2.162002455421761], [4.094330512933896, -7.7581383157545325], [-4.959005925438627, -7.964904052235324], [-1.1213158532103382, 2.9886380694994816], [-5.300589419366865, 2.7557094116562038], [7.2535139398076796, -3.478664414157417], [5.1166559205362, 7.269305514378146], [-1.0499584492886367, -0.4422789031391916], [-8.08119082999527, 2.0095643894206328], [-2.9538449449969884, 6.494380547967819], [-6.374040732236828, 0.5106896590448462], [-1.9720238751532317, 0.36530887729154693], [2.1030205609933, 8.802832905787607], [5.794712611639097, 3.737576266226105], [-0.7851117616936765, -5.533264499545503], [4.788866453232228, 2.2966260704286974], [1.4027589305030141, 5.795001973552458], [2.1263452874745274, 0.2687820350641774], [4.766830566718148, 3.5563545725037846], [6.470281231081617, -8.697451392971987], [-5.47826684098292, -7.0026426556307975], [-7.2685244968790546, 2.606715805534328], [5.867506528874281, 3.68631687809126], [-1.6164611399421929, -5.970356871971412], [6.50013328640202, -4.5664758137884025], [-3.4847394772939513, 3.970213189168084], [3.5626483622439356, 3.1468398771353154], [2.1341372713112374, -5.9804127501045725], [5.182149866785471, 3.2783076948452408], [-8.235336287469604, -2.23527555890477], [-0.9639343745329638, -7.882810032981623], [-1.4182112939452434, -2.256748839380114], [-6.73946064007812, 1.0692233572032317], [7.470602981611442, 0.3280847441390766], [8.286042755404486, -2.412630148151587], [0.41636385284039656, -0.567044489530651], [-6.104804212629014, 0.9565789234518771], [-6.188387536477148, -8.226842558495068], [3.168912639705969, -6.157840948792311], [-4.27834431417207, -0.12532981872282045], [-7.712805444709815, 4.745492454074829], [-0.13363490092978414, 2.4554996787233883], [-2.3432225764053563, 4.613498847708858], [-6.67993746964177, -7.943815952808573], [3.2204397636858486, -7.708713050366949], [-7.860201826806377, -5.057213955858028], [8.819367525209021, 5.851974892676312], [6.911502420847425, 2.8467038348925495], [7.629045052722244, -6.450483606908742], [6.042035032103646, 6.240053349848649], [4.8915388497906935, -1.62905016948591], [-7.955393872242304, -0.5509207674884489], [-2.205108621246664, 0.22704532493055507], [5.933016300116994, 6.39419320917861], [-2.5603316152314735, -4.200460534484872], [-3.237445841138247, 7.53153380052128], [2.6140937227055927, 7.979411844508564], [6.763133029562753, -7.072685916637919], [4.3550282377068985, -6.6907475937602126], [0.6080849556296595, -6.87319133342216], [-0.28883373619742314, 8.006816886815557], [-2.1499864153427355, -5.584541216489514], [-8.439299130312268, 1.4686573142083421], [1.5624208208547061, -0.4802734034562457], [-0.9071018420791366, -6.263989834693187]], "dtype": "complex128", "order": "row_major", "shape": [5, 5, 5]}}, "observable": {"kind": "identity"}, "op": "matrix_exp", "probes": [{"cotangent": {"value": {"data": [[0.03574246058286881, 0.004095890071445245], [0.1590026457944501, 0.09330335839627471], [-0.022999072771199978, 0.09088450871383597], [-0.0846680555987815, 0.06658424603022965], [0.053907330380900606, 0.07677380860943703], [0.11454761799031084, -0.01948868967177142], [-0.07609692032428002, -0.045116651477816364], [0.002976560405272444, -0.01898903376275651], [-0.07294131661400995, 0.1108673700640668], [-0.032177961237885785, -0.010760828512212262], [0.003948563817635246, 0.15474877024165], [0.043479520239282726, 0.05441311362100377], [0.06817983462289653, -0.07342177460579033], [0.009793062342864341, 0.007305144162793523], [0.0609793149977626, 0.07499053777859893], [-0.07137578888414407, 0.11695622850671285], [0.09221120284939502, -0.1632768325480839], [0.00013099552977643096, -0.05569951446352214], [-0.08380581247723819, -0.0633318193088211], [0.052638424958738494, -0.05614616197879796], [-0.0565197434931781, -0.15391184709285352], [-0.059329980414130325, -0.0060632212686103385], [-0.05462078530207338, -0.04494616689451085], [0.060352352843463464, 0.01402197827619577], [0.06395821282141914, 0.05559195064375214], [0.09465115441181109, 0.10552584262488059], [0.022637463375031787, -0.06229892231808694], [0.14890983874098315, 0.08942598894323275], [0.021635309539055004, -0.05586592111039282], [-0.02963979920931562, -0.019883106433972467], [-0.03540993105313807, -0.0560590608556298], [0.12392629829160098, -0.11280221169961283], [-0.029504791147134744, -0.015712275904379746], [-0.06223757432615512, 0.006153483597751068], [0.09972913576650731, 0.02223954587063807], [-0.008049015172995189, -0.01739897573217739], [0.09822402584879517, -0.032729735017551116], [-0.027310404026651393, 0.06008515090323584], [0.014439966905630718, -0.0004680253983574525], [-0.048332963448625504, 0.08000975362322968], [-0.01166299088786129, 0.007336420602383479], [0.0017259111584710394, 0.0007271509825585675], [-0.042600985445309054, 0.038598568433283235], [-0.06452786406565669, -0.023572466053611612], [-0.08997749494297988, 0.0524096992253277], [0.024451211425488287, -0.07132451315595541], [-0.07982761974214822, -0.09501827205007173], [-0.08547493935098747, -0.03168887660001254], [0.012068326007505617, -0.09728313144562838], [0.04323604230052756, 0.10415958611905879], [0.010181827900561368, 0.05606503672749727], [0.06792674630731758, 0.01984134058712506], [0.028308372209892484, 0.10133685447075252], [0.05145562731608096, 0.08840952062799275], [0.06947179190278409, 0.019246620219947205], [0.048731326426358544, 0.09358082034072925], [0.008165362025283472, -0.04355723142743487], [0.0439434489792423, -0.043858875406522384], [0.03408699831682903, -0.014911967352238353], [-0.08079050133193506, 0.00324178004075713], [0.020216591676275666, 0.04518814381104355], [-0.05660501556027292, 0.09577042324618312], [0.027412191245362146, -0.004653933811825544], [0.01911153275412357, 0.09686162383306772], [-0.11220731600251566, 0.10741314581139327], [0.052775679905205417, 0.022773307453782993], [0.055153516740937145, 0.017903881645271673], [-0.057692974420632016, -0.09904286381269368], [-0.017638704429499642, -0.008630557750026028], [-0.13330860017971927, 0.026517480889303727], [0.006318445476855067, 0.0199264226962193], [0.05998589400314772, 0.09048456441505277], [-0.0389013510660715, -0.04975631414573274], [-0.0235381934374941, -0.13070684694625123], [-0.0038878806931359105, -0.054721878414463056], [0.00615121338108018, 0.052880277992700654], [0.002361715939731901, -0.001162292246496015], [0.03645876956393753, 0.06398434322075407], [0.024545976293601463, 0.11417372286520447], [-0.05486850158983581, -0.03285416579946245], [0.040118544675359115, 0.010647817723257806], [-0.030348969313326292, -0.03832047810635018], [-0.05207951708688444, 0.11354533441107756], [0.08960317757836347, 0.0488061285922322], [0.019901825638624377, -0.0017017222288234857], [-0.002883149187757726, 0.01622795442543682], [4.007134212998036e-05, 0.10029511337501691], [0.05174536573986999, 0.006953549504949654], [-0.0700722128819567, 0.057195094960670954], [0.060579410121973154, 0.04393997117108004], [0.08563215301510543, -0.16345978326309046], [-0.10989855654721628, 0.0895768145529211], [-0.030885851991083842, 0.024037335413592494], [-0.00318174463025266, 0.048890385490476695], [-0.02379265113787836, -0.0863135556239709], [0.015877723185043456, -0.0930920786916939], [-0.0370222175585855, -0.06894466191336318], [-0.029517030661908552, -0.013867950095459268], [-0.0159186802635859, 0.0036715249445251697], [0.04326845002763403, -0.05782512648757225], [-0.04196355962780838, 0.04532554943626638], [0.04982419934964493, -0.0166837241768024], [0.0662156838996429, 0.04915854946877285], [-0.07334518653259937, 0.026644432924478528], [-0.03219663867369787, -0.010943259118005576], [-0.0006484291906992261, -0.013859722484837106], [0.03559029576672038, -0.046789584675954767], [-0.010265096712825121, 0.027811687396204224], [0.03972202361389095, 0.06822311614270483], [-0.03437872952229888, -0.03272265778243488], [-0.07534008123879449, 0.058080419987335936], [0.04580408652109151, 0.03223525432249903], [-0.05424471825866838, 0.12136458720153738], [0.034130085235895044, 0.04810698676236068], [0.0012624849953966062, -0.0246209778712779], [-0.006972876942191607, 0.028163170116991133], [0.07969402086236857, -0.10548821361194839], [-0.03912674011860407, -0.018597809966915162], [-0.16145728222172756, -0.052401520093143326], [0.0018729627816020463, -0.059722932309905566], [0.0014196378546418888, -0.08907960711375891], [0.013428886838095457, 0.04639443419692348], [-0.11744307619537787, -0.0451683433247261], [-0.0339260925549064, 0.0016826254049350548], [-0.13958751742897615, -0.028436861218450554]], "dtype": "complex128", "order": "row_major", "shape": [5, 5, 5]}}, "direction": {"a": {"data": [[-0.022109992671958012, 0.03791811645519793], [0.058955628133703815, -0.01846781808234453], [-0.024159428583259007, -0.038968388785956966], [-0.03646162486202826, -0.003222402277309388], [0.16397634942454817, -0.007103654781215308], [-0.018034298402419522, 0.06242866048175899], [-0.05643651848486637, -0.03348005515838912], [-0.015292104812403238, -0.04346706139962997], [-0.05905325806235551, -0.08674570245921112], [0.1011521695998809, 0.04657987247769172], [0.014411112996642952, 0.03850475388742128], [-0.024862548535439684, 0.03850606480916918], [0.12801564229438467, 0.008502659535634949], [0.05363133065558191, 0.1360908100397197], [-0.1339650823185462, -0.09066863104725895], [-0.11883022850825062, -0.004651421608422416], [-0.0025056075303311986, 0.044029392946211034], [0.08414294358832745, 0.0735966541172778], [0.10261266831570535, 0.0673709939760395], [-0.002093929195694987, 0.07741415492243879], [0.05660710108170222, -0.06275831797895035], [-0.009523879349700544, 0.0334513224642945], [-0.04859708899384904, 0.018666830328846633], [-0.0027014908207450192, 0.014827282429194042], [-0.024273437378370954, -0.04428788181334811], [0.08964836206747048, 0.025487747362347116], [-0.023935430215011266, -0.023131908200493382], [0.03561104104106648, 0.08130910568433272], [0.0029360992782605814, -0.04411222422479182], [0.022349156711674877, 0.032342358511471055], [0.0041859511372844185, -0.00765777498478834], [0.05882683793050901, -0.007055882227991031], [0.0432849793978411, -0.04313170141588078], [-0.004321735337384198, 0.012686451677455942], [0.07033317218070335, 0.047332816820802676], [0.10905582085345665, 0.06699460638853307], [-0.03354554163008634, -0.02753108231758771], [-0.03864965514461008, -0.07176451125644394], [-0.07178138674526645, -0.005008039033052196], [0.13273662784196702, -0.011080793177838718], [0.08593922752635147, 0.016463022250601086], [-0.16101922269047025, -0.09462346547731142], [-0.01814659749720909, 0.054761657014047074], [-0.00769634557397791, -0.11478657945306395], [-0.0009837151134031452, 0.04749610571752718], [0.027337973452096282, 0.10462847583432704], [0.06988318021229047, 0.01478998669073919], [0.016557950190617173, 0.03656924038414564], [0.08516850475639727, -0.0242689264620653], [-0.049995950858906686, 0.036766125539705194], [0.02376110479448772, -0.040521171034826366], [0.13657623729676396, 0.037047207907070584], [-0.01441010133561181, -0.013097814696660371], [0.07548083927149692, -0.004811839085549848], [0.007485670957061792, -0.0869460879197399], [0.00986028901673946, -0.11934195154907551], [0.010869223367947599, 0.12459545775467502], [-0.11197872153479266, 0.04621285070635841], [0.03872031261542508, -0.01788563068841203], [-0.09265728774116307, 0.00879733081831268], [0.03344417818679031, -0.035541940601904166], [0.026927008643893573, -0.11650496393544756], [-0.057404707844185554, -0.04259754371581491], [0.06636558667439826, 0.023242492184518418], [-0.07547301589403264, -0.054313686315686664], [0.09901424020096969, -0.017745222237455296], [0.06021306698307081, -0.07783599979758256], [-0.1290418550564689, -0.025334180968916212], [0.0467859446489582, 0.05710832220292487], [-0.030477399028543486, -0.03196611743912187], [-0.005149660055381184, 0.044826031987458236], [0.00869281802442266, -0.024505217832316246], [0.0727065304849153, 0.06481901642761362], [-0.030554619737643195, -0.07986757162996205], [0.02471732045060984, 0.018823226784089528], [0.06893800231986914, 0.07972206658123573], [0.05797547366369295, 0.09116016323293397], [-0.004977407048279002, 0.01569903870882875], [0.08026605222300295, -0.06107503330787242], [0.11240101214775887, -0.11464949890083302], [0.038998313403909884, -0.12943071934795009], [0.018290990762972564, 0.029571793726747526], [0.13001920359599028, 0.0073056707610061385], [0.07625770455977152, 0.012432363379863988], [0.10624138827862457, 0.03474954194529854], [-0.02857501998864004, -0.05068703055407262], [0.023412364299120247, 0.005962425827914773], [0.02276572552338952, 0.012066232999677561], [-0.059024160038474625, -0.12942800121328768], [-0.007646287010046954, 0.1612275043872847], [0.0059706251032708366, -0.018026401513442272], [0.05671052464744432, -0.004756334333961996], [0.0010085995153050872, 0.01487440151434297], [0.0031695812828252666, -0.006175040708815181], [-0.051685650272772415, -0.025874080401645554], [0.12129295826527552, -0.12717330533787627], [0.0362005345835475, 0.10094892612555903], [0.01693000074083398, 0.013278094062319367], [0.02860265625074113, -0.012701326338721392], [0.11424616425720666, 0.034437540869309426], [-0.07977064541059214, -0.04246735583777237], [0.17350157518178064, -0.05366109025289825], [-0.056287816515033756, -0.018975399361588558], [-0.026390763536869605, 0.025859572516877007], [-0.09488853704285531, 0.08606214696245007], [-0.04349327829817318, 0.06278517763052001], [-0.08398297257789794, 0.0380707834701855], [-0.0025818008785431228, 0.025410485041582026], [0.126584737150813, -0.0653922863535909], [0.06142938040964131, 0.06123642107376505], [-0.026830719140320574, 0.0023745947004029434], [0.10760003050126402, 0.03497927855738527], [0.05695277341044159, 0.013285417020566416], [0.08754832448234694, -0.07351993264785248], [0.003784131064400202, 0.020518022356203793], [-0.0005094758810587643, -0.08466899167100661], [-0.023767400144074, 0.058735282277385475], [-0.014161353808856079, 0.07771475934349881], [-0.007149792066158037, -0.00033393308826493894], [0.009768699085467627, 0.017387083793791152], [-0.07144186090885928, -0.026514066770939763], [0.0454711803990232, 0.12148833939176377], [0.03943587084183971, 0.14339627978216277], [-0.044353866284309505, 0.0028756082578308643], [-0.026423474622299693, -0.02815173555470151]], "dtype": "complex128", "order": "row_major", "shape": [5, 5, 5]}}, "fd_ref": {"hvp": {"a": {"data": [[1.0104811466546197, -1.065450528437167], [5.14161356124783, 1.6160136196406139], [-2.4708320928350256, 1.4022871836412885], [1.5124805901756417, 1.9280813241560868], [0.076025268051252, 2.732685085649549], [-1.5179147419879973, -1.7293675482968776], [2.8248787197768244, -8.439984507825029], [2.1929272645052915, 4.045638377517536], [3.171332710327861, -2.3120343971212947], [4.463289835402017, -0.15121020036131344], [0.8681289109105128, 0.38720165944857804], [0.19245211520923347, 3.299731482428476], [-1.3293042540295288, -1.1558315363849727], [-0.8062702979153721, 1.3723252597898876], [-1.4436652304412012, 0.7035990911174931], [-0.2398443909464342, 0.1696606248119887], [-0.9439725453696987, -0.23089519090471144], [0.4199564276559936, -0.27138453555219677], [-0.22739229518312856, -0.28625210323104755], [0.04622192024999088, -0.398225225892774], [-0.9258340866163254, -0.7236858645146722], [0.790829318675706, -4.243269873215906], [1.3366558334059522, 1.8394624817591076], [1.413343666211282, -1.407381521246984], [2.1363655468758744, -0.3860637289236454], [1741.7662738906442, -14872.212599945087], [8143.579950424631, -44.73589150049954], [3467.114638050069, -1175.92514743023], [7702.263688257519, -15997.196475403112], [17695.1891595591, -4849.503003233636], [-13572.160183612523, -5561.135304790306], [1780.1682892463332, -7849.609426606905], [-80.48673695933806, -3896.1474521468595], [-13580.36072667301, -10812.744058351316], [1022.8749323624118, -18604.56830048481], [613.5004480568995, -25591.070209906775], [13917.77791496134, -1563.2803269955132], [5934.81024270293, -2670.5931075738354], [9963.834502587099, -28398.069791343965], [30224.301615518543, -11480.030299918963], [-12440.089576441851, -19252.53160635133], [9779.854037149473, -7852.53680623121], [2805.7159300241997, -4875.847713185885], [-5894.911778047562, -26166.948601523578], [15064.186910700366, -24320.93807737398], [-46449.315488034474, -22115.927353677078], [8428.394383209625, -26376.77577121286], [-252.83088430504026, -13212.666570314621], [-43202.320038268015, -40965.933000758894], [3467.176567675645, -66321.24408573932], [-0.013469847025163071, -0.01051851048218654], [-0.021589270038375873, 0.024383180666493445], [0.0012345179162017888, -0.005950051956441378], [-0.005985837532133747, 0.0034369254163310156], [0.0008750328159759304, 0.004576348252150787], [0.037514853520682956, 0.004606796903409736], [0.019927431966678538, -0.07452007369253852], [0.005990352901192558, 0.017014213499118372], [0.010055662165221524, -0.013677710643636347], [0.0003234778529311436, -0.005629686320871658], [0.009334786535989768, -0.005119917034776544], [-0.009559350768570363, -0.020709054409725647], [0.004898837091259465, 0.0022303069019262486], [-0.0006538621419342284, -0.004820040453946478], [-0.0024302987977884525, -0.004044229342426412], [0.030926560607006437, -0.0056714310855844625], [-0.0003687363904431806, -0.062348453046686875], [0.006994700802644198, 0.01280106601679431], [0.003764241883013357, -0.014102003742916388], [-0.0010346903669816011, -0.0026982272726094694], [0.02532210032652627, -0.0021502253497801263], [0.003402221886978803, -0.05098213583540586], [0.004128979870648466, 0.010199955503156936], [0.0029420789233280706, -0.01193569225706772], [-0.0005117190066158882, -0.0036135287214499492], [568.5487110745652, -558.0019532149738], [227.64251708945883, 582.9135660179144], [141.22599380973747, 1.605058387275242], [787.3182483097081, 370.11705807085906], [105.0030683820808, 195.9127427424238], [96.55824412543977, 309.2228438231999], [-207.85456614875457, -24.266060024243494], [-78.06832411465676, 9.192060743527088], [-224.56913889738576, 163.59022917853056], [-20.577451107469383, -6.7791778870991255], [162.3847658793013, 1472.542447447285], [-1022.6501047555945, -299.285753544745], [-209.76136443286623, 110.95250159759536], [-1322.0272431688547, 688.4088879064775], [-291.7854639247043, -27.948902206142495], [854.2486516325836, -782.4422655414552], [315.78650885742917, 761.4248618199762], [109.9006007766081, 71.8739143263817], [1062.9082217992836, 457.6755502600321], [166.47999430376615, 157.6649858091781], [509.8630980178235, -974.9237237370176], [497.16601559016914, 618.1327315312427], [192.58236924129187, -21.74900683879427], [1134.6675059757345, 148.51450451787403], [183.33453442711348, 169.00008146207264], [500.0953965902576, -2098.32136678795], [-331.76671406866376, 5167.7018314138395], [4728.754022309808, -1537.2265949838588], [2703.882228840522, 3093.7411777536067], [-2955.7396250306774, -3182.0355559354734], [-5654.51030098605, -300.1189722553612], [-1892.4855706696576, -10671.171088478068], [-11590.466207417454, -1065.4706438276924], [-8447.484756540362, -5709.739756020667], [1121.0787133126707, 9431.068168306288], [-2021.757683681799, -1985.808635064739], [6194.9828949517905, 2635.343136587228], [562.3806965708226, -6568.605794287016], [4840.0581001768405, -2068.3029111372175], [-5513.315872910952, 1771.347290136604], [-675.3285692470297, -503.9616293691167], [13958.708353202035, -8351.78169816835], [-5883.470752622242, -9503.036156087082], [4501.835460632909, -12979.0949629625], [-4487.502124255301, 6588.689633109332], [1489.554929006114, 1032.2522598361777], [-7229.047930494856, -2431.960060630362], [-734.9292687459352, 6293.056538399047], [-5833.7100963666735, 2413.8102930236587], [4842.130867105793, -1713.4509049590736]], "dtype": "complex128", "order": "row_major", "shape": [5, 5, 5]}}, "jvp": {"value": {"data": [[12.624263828306123, -1.7805834604456139], [5.685872134854798, 27.71333507858209], [3.781692712703861, -8.651664096123913], [-3.8068292907950627, -2.653965078774883], [0.04851027366613701, 11.787660750991703], [5.246385963865797, -39.15401140177062], [86.47413990952334, 5.678951719327375], [-23.153032752692518, -17.01286139023132], [-11.064424546733665, 10.11946604923763], [34.937891455467785, 9.037477393356676], [-17.681490930409144, 12.717258108421175], [-30.914368211132665, -36.79388913247068], [2.112541549919501, 16.34941341212995], [8.478865698800684, 0.9443606995696843], [-9.82571813816627, -17.618519407365092], [-8.865328582310184, -18.163504810317935], [39.40567477037144, -22.586696205987042], [-15.653804467252403, -1.879897468155155], [-2.6893761562645873, 6.967222837727353], [18.160624466615882, -5.877359124173952], [-19.791742749741157, -9.34040471052202], [18.32897730281921, -44.29951604470601], [-12.542300054886535, 8.101229032136876], [3.3777370962090023, 6.738397660800381], [10.909025892639244, -16.366870211166464], [-117310.92962287573, 40410.95046213643], [-111269.42707908803, -76278.98997195695], [-212978.29643363258, 77449.93781832178], [-183279.64017716318, -110549.38443690036], [-355961.41286370205, -376041.5399682015], [16467.119375931867, 74322.39794990067], [-56286.12482733443, 63964.94149120444], [31654.450150392837, 137855.07226085887], [-83971.46465096659, 102463.1915679834], [-262696.51913747133, 191822.92039159825], [3763.1679379598077, 27411.491963648255], [-23786.48024528235, 21300.820918603993], [5754.772909988122, 51692.63945983706], [-34558.25957266859, 32831.5475405925], [-104772.56286649562, 58842.252041830354], [-128394.2169923649, 100098.2179822216], [-170114.44828152898, -48132.90781535876], [-231391.26914084653, 185441.0221249255], [-273748.2115458779, -59478.48134584475], [-599655.6570413173, -316450.4722609197], [-5650.504795239009, 123821.14747130632], [-120367.27744234263, 75466.39733862142], [-14176.91822023654, 232995.23821463174], [-176578.43566358645, 118865.56735817358], [-505301.6240074307, 177858.11864547408], [0.22038646542996074, 0.019063777205367536], [-0.4484079088732362, -0.3089440299775055], [-0.0692446947074783, -0.117862203743414], [-0.2742898260127435, -0.3116302572886335], [-0.2802949011841998, -0.2742558721172295], [-0.045774650182740795, 0.3810563840474673], [0.4467435760759711, -0.8517242513753039], [0.23784302784745276, -0.1272727104067027], [0.4647463350775999, -0.5388244999464802], [0.4273965710498283, -0.5232343617953722], [0.018922087595437013, -0.07619991285619473], [-0.22454166260096708, 0.16279480029849275], [-0.03411278894524204, -0.002866445374819791], [-0.18905979907136847, 0.10322555873624964], [-0.15504054351585148, 0.10678914172048974], [0.017605264551697388, 0.0175445772084158], [-0.10368406445121303, -0.10063590885760418], [0.03627917587532317, -0.03290708895042623], [-0.03169847835286542, -0.07886339071789991], [-0.02085174983021754, -0.054328551793613976], [-0.04606760964435306, 0.021096022680999936], [0.02463489209923139, 0.06886343022745005], [0.039797669142714626, -0.0062610916302113006], [0.03452152038998519, 0.07065211933399478], [0.05915906498319353, 0.056721740241585626], [2964.0363829557436, 1017.0918488959305], [1483.7466737775064, -1998.6382738737173], [407.1666760694499, -8417.935114601472], [7078.355736953519, 3670.0994834595126], [3321.2498748499006, 4208.206269783298], [647.5826949113765, -2766.3652348620567], [-2008.3077435948042, -493.8267415051541], [-6955.82888464671, 1838.0284623601513], [1394.2100602912355, -6228.126038551979], [2755.3506992726284, -3801.4721502295447], [506.3424099713527, 1055.1003513920523], [-278.82624565876216, -1090.7305766774941], [-840.9288708167348, -2806.5688920833254], [1868.2151081002858, 201.3101988387258], [1456.3610464224769, 1689.9097323321594], [3840.431306333088, -2349.682230025897], [-2098.285760819807, -2063.8593072720205], [-9939.058101246048, -4291.62027419633], [7358.9293798050585, -5761.585997477009], [7379.024976176909, -1872.8464137969513], [1442.2921612871162, -1660.3083785306646], [-1110.857392254952, 236.64111963471845], [-4004.4967852737136, 983.1922206673885], [1162.246712175734, -2854.8340470404937], [2237.956812704281, -2364.6215426468702], [15552.794337969186, -32110.20417548344], [26625.2672989898, 93592.88979053103], [49543.3352024778, 7310.877875627508], [38454.747306418, 21929.239861561262], [-31503.0184110587, 3311.517579239139], [-39972.46274495635, 53650.6757699875], [59300.19589791772, -110712.53524387418], [-80495.96021480835, -29092.649441053527], [-46289.51513861889, -135014.44553867812], [73363.82347410623, 21264.379002669026], [-42405.301464072516, -52357.07106040384], [145774.42263233758, 83284.3399261208], [52605.982904780285, -73650.92454211612], [120033.16920122119, -45839.60327422297], [-28037.376201725976, 66510.18312536432], [-49095.782120255004, 19862.84621883518], [88464.3899959664, -46567.48657828339], [-41697.61600314334, -53153.8479361033], [26341.340301297892, -113712.08143074485], [41089.3238890831, 45075.304941710216], [60989.84743551887, -12319.64683999988], [-124712.65768769252, 94726.71398783503], [38912.20202378701, 74781.97575334036], [-8362.652987594269, 109440.52604040889], [-41641.78044153169, -49197.03158758043]], "dtype": "complex128", "order": "row_major", "shape": [5, 5, 5]}}, "method": "central_difference", "stencil_order": 2, "step": 8.262191778179044e-06}, "probe_id": "p0", "pytorch_ref": {"hvp": {"a": {"data": [[1.0104811095494788, -1.065450532528569], [5.141613496814368, 1.6160134925863452], [-2.470832031480695, 1.4022872052379416], [1.5124806103490114, 1.928081265765392], [0.07602532060954831, 2.7326850266637597], [-1.5179147594949494, -1.7293674963321894], [2.824878515390062, -8.439984428039642], [2.192927305241781, 4.045638287815052], [3.1713326297219013, -2.3120344370214596], [4.463289755272624, -0.15121028812838583], [0.868128915321114, 0.387201632793054], [0.1924522017364238, 3.2997314338777004], [-1.3293042671866486, -1.1558314933390963], [-0.8062702578790022, 1.3723252739551108], [-1.4436651885402103, 0.7035991263384631], [-0.23984437770762654, 0.16966062506675408], [-0.943972515022735, -0.23089516071796948], [0.4199564090761581, -0.27138453654675965], [-0.22739229828398413, -0.28625208527094587], [0.04622190976185983, -0.3982252053473689], [-0.9258340913969136, -0.7236858376862264], [0.7908292232719409, -4.243269825292506], [1.3366558494096907, 1.8394624362227605], [1.4133436240479131, -1.407381536945763], [2.136365504096541, -0.3860637681158935], [1741.766318884762, -14872.212651268674], [8143.579981798181, -44.7358700898269], [3467.1146516718236, -1175.9251472100113], [7702.263781589081, -15997.196508984398], [17695.189230010907, -4849.502987004294], [-13572.16021154384, -5561.135379390575], [1780.1683269959742, -7849.609449148154], [-80.48672808589897, -3896.1474686637785], [-13580.360713096867, -10812.744171510993], [1022.8750018024954, -18604.568363950588], [613.5005445308551, -25591.070340124134], [13917.777992785028, -1563.280285751255], [5934.810276634371, -2670.5931114612185], [9963.834709863955, -28398.06989047446], [30224.30179153828, -11480.030276941648], [-12440.089549748202, -19252.531792554524], [9779.854141003045, -7852.536802937405], [2805.715958841124, -4875.847730678428], [-5894.911627569304, -26166.948822130325], [15064.18709637179, -24320.938147573856], [-46449.3156063745, -22115.92764866945], [8428.39452920823, -26376.77585522818], [-252.8308610602963, -13212.666630891526], [-43202.319992011544, -40965.93345447477], [3467.176790876126, -66321.24435393531], [-0.013469846976614665, -0.010518510493891272], [-0.02158927006778223, 0.024383180523109872], [0.0012345179514611235, -0.00595005194900907], [-0.005985837543217654, 0.003436925367288705], [0.0008750328064648015, 0.004576348227508105], [0.03751485345507994, 0.0046067970100531295], [0.0199274322816395, -0.07452007344361342], [0.005990352823406944, 0.01701421351997842], [0.010055662300228172, -0.013677710547469614], [0.0003234778983483912, -0.005629686246455183], [0.009334786545563386, -0.005119917008862869], [-0.009559350674141239, -0.020709054411444383], [0.004898837065694311, 0.002230306924050735], [-0.0006538621092216895, -0.004820040471780618], [-0.00243029877277742, -0.004044229334068677], [0.030926560593819742, -0.0056714309623182675], [-0.00036873605691355865, -0.062348452953771366], [0.006994700750145492, 0.012801066051817942], [0.0037642420233416794, -0.014102003692454837], [-0.0010346903208582922, -0.0026982272342869944], [0.02532210032162652, -0.0021502252769464414], [0.003402222072068608, -0.05098213575941353], [0.004128979836950697, 0.010199955517400265], [0.0029420790233732255, -0.011935692214117095], [-0.000511718989606439, -0.003613528696287895], [568.5487161800747, -558.0019532331906], [227.64251502842689, 582.9135692323097], [141.22599491528604, 1.6050585647332412], [787.3182513782824, 370.11706370346076], [105.00306815843851, 195.91274434413043], [96.5582442456836, 309.2228450719358], [-207.85456684144896, -24.26606041515509], [-78.0683240854139, 9.192060867181183], [-224.56914011999118, 163.59023002397083], [-20.577451343603673, -6.7791778095802355], [162.38476336536905, 1472.542453507065], [-1022.6501074462504, -299.28575741643397], [-209.76136487564122, 110.95250267131175], [-1322.0272513000957, 688.4088891069732], [-291.7854657150932, -27.94890292097203], [854.2486577493731, -782.4422695086487], [315.78650881743556, 761.4248669112734], [109.90060209410052, 71.8739135837304], [1062.908229657182, 457.67555494001556], [166.47999536119818, 157.66498776228167], [509.86310236015277, -974.9237270700269], [497.1660158734103, 618.1327352005619], [192.58237011496783, -21.749007364451046], [1134.6675119305266, 148.51450735468478], [183.334535288881, 169.00008277486822], [500.09541117543745, -2098.321349748788], [-331.76669174055206, 5167.70179658849], [4728.753997419773, -1537.2265863841885], [2703.88221158055, 3093.7411414682238], [-2955.7396029072515, -3182.0355443501776], [-5654.51034709525, -300.1190781085985], [-1892.485720521322, -10671.170929087675], [-11590.466070164279, -1065.4706509448274], [-8447.484742402125, -5709.739554262518], [1121.078610136562, 9431.068079237395], [-2021.7576354341413, -1985.8086508448087], [6194.98282842362, 2635.34306503842], [562.3807002092408, -6568.605732862183], [4840.058015084083, -2068.30291107978], [-5513.315829894928, 1771.3472483531614], [-675.3284850434756, -503.96171879307616], [13958.708142847998, -8351.781759609788], [-5883.470689026184, -9503.036031410054], [4501.835276766985, -12979.094872511634], [-4487.502081825947, 6588.6895000694185], [1489.5548945051646, 1032.252274781056], [-7229.047883049541, -2431.960016353252], [-734.9292697031376, 6293.056496988256], [-5833.710032215198, 2413.8102936154937], [4842.130840474507, -1713.4508735859522]], "dtype": "complex128", "order": "row_major", "shape": [5, 5, 5]}}, "jvp": {"value": {"data": [[12.624263716224874, -1.7805832969591053], [5.685871770860341, 27.713334902114184], [3.781692834923038, -8.651663972332564], [-3.8068291949561877, -2.653965102440888], [0.048510127776669076, 11.787660647954125], [5.2463863194830465, -39.15401090395906], [86.4741389969447, 5.678952574241668], [-23.153032259089155, -17.01286161463295], [-11.064424513632888, 10.119465743216482], [34.93789100273422, 9.03747770861679], [-17.681490890264655, 12.71725780172996], [-30.914367585646524, -36.79388913967494], [2.112541292028922, 16.349413325979004], [8.478865579860907, 0.944360801030204], [-9.82571786313627, -17.61851936965345], [-8.86532828324241, -18.163504749748526], [39.405674731609295, -22.58669558325127], [-15.653804365462314, -1.8798977152943734], [-2.689376244655269, 6.967222713252717], [18.16062440879731, -5.877358860013187], [-19.791742394328573, -9.340404789331616], [18.328977562087065, -44.29951536583245], [-12.54230006368774, 8.101228712745066], [3.3777369392858594, 6.738397570348016], [10.90902595960961, -16.366869894441173], [-117310.92994806409, 40410.95066617679], [-111269.42767695758, -76278.99014292282], [-212978.29730247922, 77449.93839230205], [-183279.64116553913, -110549.3848028236], [-355961.4147900916, -376041.5412147405], [16467.119448140824, 74322.39816027573], [-56286.12499401787, 63964.94182120183], [31654.450361257117, 137855.07281528114], [-83971.4649498353, 102463.19207758455], [-262696.52006843087, 191822.92135795896], [3763.167948092105, 27411.49202623885], [-23786.480308569367, 21300.821004275575], [5754.7729418937815, 51692.63962437062], [-34558.25967145576, 32831.54767524787], [-104772.56316242492, 58842.25226382086], [-128394.21726959608, 100098.21841702648], [-170114.44916423626, -48132.907657294105], [-231391.26988144324, 185441.02336010477], [-273748.2129763793, -59478.48123846169], [-599655.6602834936, -316450.47259229346], [-5650.504799879742, 123821.14779890483], [-120367.27782752046, 75466.39769780771], [-14176.918206806844, 232995.23906762977], [-176578.43628414907, 118865.56790489586], [-505301.62574435305, 177858.11947176966], [0.22038646514648472, 0.01906377519679205], [-0.4484079116003011, -0.30894402567158025], [-0.06924469519721228, -0.11786220322953248], [-0.27428982869548196, -0.31163025456787763], [-0.2802949034050753, -0.27425586986561945], [-0.0457746477499822, 0.3810563832682851], [0.4467435676706778, -0.8517242571437547], [0.2378430279658711, -0.1272727110845917], [0.4647463283244166, -0.5388245044440383], [0.42739656592799535, -0.5232343655046408], [0.01892208738045607, -0.07619991244816851], [-0.22454166260376968, 0.16279480140679653], [-0.034112788992038845, -0.002866445764367581], [-0.18905979863656114, 0.1032255601020423], [-0.1550405432826264, 0.10678914264707562], [0.017605265643463524, 0.01754457605837042], [-0.10368406693221587, -0.1006359098789853], [0.03627917544253437, -0.032907088483755995], [-0.03169848049285039, -0.07886339246796835], [-0.020851751508648663, -0.054328553061397805], [-0.04606760950848459, 0.021096023130932406], [0.024634890866543316, 0.06886342869764638], [0.039797669498207774, -0.006261092128133779], [0.034521519457167416, 0.0706521187014439], [0.059159064262659614, 0.05672173960160826], [2964.0364156963315, 1017.0918334695077], [1483.7466567778006, -1998.6382869057356], [407.16661364581705, -8417.935140158133], [7078.355788624123, 3670.099460127412], [3321.2499233159656, 4208.206267263639], [647.5826709086759, -2766.3652535567962], [-2008.3077474486574, -493.8267273992886], [-6955.828884959605, 1838.0285125028793], [1394.2100298195073, -6228.126071771463], [2755.3506814985444, -3801.472182741333], [506.34242191890684, 1055.1003444518292], [-278.82624555043117, -1090.7305776463568], [-840.9288789495017, -2806.568899386922], [1868.215123461216, 201.31020226726542], [1456.361056763337, 1689.909732208617], [3840.431310468823, -2349.682277786954], [-2098.2857797034508, -2063.859304468194], [-9939.058165277402, -4291.6202423518125], [7358.929407188157, -5761.586063223154], [7379.02500623176, -1872.8464604621106], [1442.2921519999436, -1660.3083918902369], [-1110.8573962391233, 236.6411201632028], [-4004.4967979289154, 983.1922346867588], [1162.246713068229, -2854.834070365399], [2237.9568137186566, -2364.6215574712305], [15552.79410348336, -32110.20445007616], [26625.26858168839, 93592.89079803161], [49543.33549081092, 7310.877420036425], [38454.748314095836, 21929.23899971888], [-31503.01856644736, 3311.518042336487], [-39972.46208536379, 53650.675812916714], [59300.19290714324, -110712.5348208875], [-80495.95997817253, -29092.648429548582], [-46289.51537238018, -135014.44302956658], [73363.82303551633, 21264.37820202202], [-42405.301739729985, -52357.070733383], [145774.42357859816, 83284.33821023992], [52605.98229962693, -73650.9248143322], [120033.16799257603, -45839.604336409626], [-28037.375605509726, 66510.18321189909], [-49095.78159327317, 19862.84655933622], [88464.38746120845, -46567.48751664307], [-41697.61626034816, -53153.84704023081], [26341.3390584234, -113712.07959133942], [41089.32388066302, 45075.304067575526], [60989.84736495264, -12319.647266452957], [-124712.6569634781, 94726.7157109341], [38912.202591529436, 74781.97547493516], [-8362.651410468668, 109440.52580818027], [-41641.78088285762, -49197.03120742946]], "dtype": "complex128", "order": "row_major", "shape": [5, 5, 5]}}, "vjp": {"a": {"data": [[-21.38203552006115, -2.087254844621568], [-31.472824728616743, -60.78214405036806], [32.145848707793796, 11.536691438031987], [12.22850197232249, -32.22963081703556], [28.229414507096905, -30.335802615908786], [-10.609301208448063, 29.845928619180388], [-99.37586833033359, 34.16832024106052], [23.858128755947686, -45.45314074283511], [-43.35317647203137, -25.5564318955359], [-38.38669583176006, -48.72604095708369], [2.628518766813201, -16.29997925455471], [45.01515213016967, -23.993787285529997], [-8.375895499110012, 23.57729021960105], [23.395623052647075, 9.683400300272925], [22.25502078032285, 20.95366601401325], [8.954933700839891, 0.4174319137352997], [15.739885880596976, 19.22464263259211], [-11.792585141021135, -2.02287751632285], [-3.3660873448890762, 12.055400665168396], [-8.384948918477422, 12.316548125218132], [-3.498941166599496, 16.48899366673694], [-49.79791051511046, 22.10310886870974], [10.501180012803154, -24.7528661169401], [-24.278440168664826, -11.348293521025175], [-22.35677919533999, -23.73171326939784], [135681.66148252029, -35219.21722256573], [27883.917264103206, 68437.61445597096], [19811.01271408826, 25392.22813382854], [169901.69346781197, 14441.368783006303], [82502.8557265033, 129922.90850028508], [31909.256597629395, -182801.6760561605], [96119.41546015635, 1801.0588245939562], [44832.49949574872, -10972.05097367987], [105068.12584478967, -194223.63472226757], [211230.75015161108, -20464.82443661175], [302134.13752693764, -106179.29678540297], [76020.75519401664, 149696.7708829445], [51057.49693788779, 54517.92630080343], [386764.7727725954, 1007.5658340892296], [214816.0248311123, 285148.5245777151], [108652.4596725709, -252950.769495169], [136916.47784261202, 43639.27299793142], [58292.618977286824, -4288.482052492613], [234076.8128229243, -238649.36783608573], [305375.6094805052, 8985.851709160135], [38899.234588241605, -676732.7675770222], [348494.64464164706, -26239.949176246697], [148242.24778590738, -62721.11520631948], [315204.1922023725, -743912.3280063765], [751295.0635692875, -196770.2502756318], [0.0368053381429708, -0.23870917734316627], [-0.540916750257689, -0.21832788007984816], [0.10887557323110601, -0.039316598022929715], [-0.22884890299315044, -0.05647600722017283], [-0.08750188534306101, -0.08173707257286224], [0.3042992686703821, 0.4776753365515771], [1.2996324614831274, -0.5861238711584009], [-0.05591840149065015, 0.2363886977868423], [0.5108252244435788, -0.2732115576811625], [0.3397111110281525, -0.06539350088206447], [0.12728232600729775, 0.051677865755885705], [0.15305886888611311, -0.21075014438776776], [-0.011674240482500837, 0.04562508343447295], [0.07660601376006065, -0.09027343164061458], [0.008184475397534413, -0.020733445347470517], [0.3184094457612442, 0.32133107352639045], [0.9351023584300054, -0.6591007804971569], [-0.006963607902449731, 0.21185629245731402], [0.34880466991882336, -0.27397596773337385], [0.25693614192052633, -0.0899563711355164], [0.19157275987204425, 0.2764520611898794], [0.7819930284022472, -0.3683421759190695], [-0.033113934769021365, 0.1550340501477665], [0.30895359274728584, -0.1557735392880562], [0.2230382255113492, -0.0341608722614509], [5545.251819958986, -2272.5890815497332], [-916.2751905011934, 4372.6159521857435], [1948.4908161134722, -312.47616742428704], [5028.937935068353, 5867.099812274453], [67.11570537644457, 2363.555714334691], [1445.5743381374866, 2488.8295109503188], [-1942.830544284272, -169.13922379911946], [-52.05170230096936, 495.382158085823], [-2013.9949608350025, 2266.2665498155434], [-590.3682933308614, 81.71608152218096], [277.76748613902663, 11549.893258934851], [-6962.030857712776, -4488.537116570422], [-877.5821792616027, 2797.2574100298657], [-12822.94166498553, 4695.342958315896], [-3418.258768662964, -1151.5776079687498], [8357.006122512716, -6275.541888312318], [412.0820766767394, 7356.738141937112], [2164.6564950351167, -898.3231040364014], [10235.12846767153, 6571.600057425944], [1167.8653109221082, 2822.734472470499], [4005.2310773889253, -6435.949314471651], [2227.6436850522773, 4963.89823980559], [1512.6380060423048, -1298.163960412737], [8805.032573531427, 2082.6945264038354], [1518.611096547665, 1910.4344412480561], [-75463.28824510555, 47283.95262847553], [155973.36524395528, 55701.11805142951], [-50204.05179586607, -92675.31881143233], [118234.53934185684, -32808.55012945858], [-45555.933238061116, 100110.73628308937], [366639.6184646225, -76647.76175118894], [-555045.8875105609, -439285.41305962775], [58509.54580570798, 442756.77752669284], [-522819.8848693756, -42278.593066437556], [327669.2967617664, -325082.16103977355], [26011.86192024683, 128148.50173051338], [157995.0708991602, -192943.45556167993], [-153721.2238596453, 19305.62860762068], [17062.946630641174, -181898.30682564128], [112776.57987194692, 114400.00412166737], [124781.87762721124, 270715.04681793804], [236497.53755601862, -515825.25910718075], [-335085.9691058268, 123450.64945151242], [-61976.20752322909, -415933.85191344854], [309643.9132222074, 197554.4295691544], [11330.484771226265, -126268.82294483492], [-193910.39061133616, 133324.3212425369], [149677.37944257952, 23652.28865039094], [-63742.676581508524, 164390.0487490445], [-73596.65494040889, -138638.52305216572]], "dtype": "complex128", "order": "row_major", "shape": [5, 5, 5]}}}}], "provenance": {"fd_policy_version": "v1", "generator": "python-pytorch-v1", "seed": 18, "source_commit": "449b1768410104d3ed79d3bcfe4ba1d65c7f22c0", "source_file": "torch/testing/_internal/common_methods_invocations.py", "source_function": "sample_inputs_matrix_exp", "source_repo": "pytorch", "torch_version": "2.10.0"}, "schema_version": 1} +{"case_id": "matrix_exp_f32_identity_001", "comparison": {"first_order": {"atol": 10000.0, "kind": "allclose", "rtol": 10.0}}, "dtype": "float32", "expected_behavior": "success", "family": "identity", "inputs": {"a": {"data": [-8.478373527526855, -1.765825867652893, -4.3228044509887695, -2.400455951690674, -7.950586795806885, 3.611605167388916, -8.067646980285645, -0.5734938383102417, 3.1285104751586914, -3.033684730529785, 5.106744766235352, 1.135137677192688, 4.947308540344238, 5.774446487426758, -3.973021984100342, 3.270799398422241, -3.893850803375244, 2.8211474418640137, -4.702395439147949, 4.163121223449707, 1.822007417678833, -3.522088050842285, -4.414461135864258, 2.3285136222839355, 8.397199630737305], "dtype": "float32", "order": "row_major", "shape": [5, 5]}}, "observable": {"kind": "identity"}, "op": "matrix_exp", "probes": [{"cotangent": {"value": {"data": [-0.19751792644609123, 0.246622019704583, 0.23916412282687893, -0.18698699297573082, -0.08950254838897506, 0.38341858009489055, -0.19809041414346143, 0.06123155572507452, -0.10478136422417013, -0.053940585376115735, 0.15522512399403707, 0.2956142350476721, -0.08323922541978575, -0.19154974331859387, 0.39464514445560034, 0.1715917990745668, 0.07322048209955567, 0.09974527850800313, 0.12753443096106426, -0.16496118484301694, -0.16285089734948843, 0.38484679675441175, 0.05274601146760097, 0.12555797079281966, -0.0620407967907714], "dtype": "float32", "order": "row_major", "shape": [5, 5]}}, "direction": {"a": {"data": [-0.23585369033065898, 0.27740903336303335, 0.42071942423919895, -0.07744624833166121, 0.015228160845849001, 0.10102743907933207, -0.13494628009889123, 0.05668988174965219, 0.04764131969392213, -0.3917451309893902, 0.29166571864015, -0.03746070157632864, 0.022137804740163535, -0.05046897135478265, -0.04172766377730867, -0.09687616203764014, 0.03424753081714751, 0.04889197036665491, -0.06829506570908277, -0.13306453175275604, -0.10417007147015098, -0.4128537274772358, 0.015031920944397046, 0.4010950727408621, 0.1668705411852566], "dtype": "float32", "order": "row_major", "shape": [5, 5]}}, "fd_ref": {"jvp": {"value": {"data": [910.1663818359375, 2097.40234375, 4259.4677734375, -108.84281921386719, -8616.107421875, 755.2175903320312, 1640.7957763671875, 3686.57568359375, 69.49476623535156, -7127.837890625, 2351.79931640625, 5514.59130859375, 10846.880859375, -440.5037536621094, -22266.08984375, -342.41192626953125, -854.8329467773438, -1502.3079833984375, 146.30670166015625, 3256.263671875, -3436.391357421875, -8060.671875, -15853.9951171875, 643.7640991210938, 32547.08203125], "dtype": "float32", "order": "row_major", "shape": [5, 5]}}, "method": "central_difference", "stencil_order": 2, "step": 0.023355829238891603}, "probe_id": "p0", "pytorch_ref": {"jvp": {"value": {"data": [910.0888671875, 2097.26806640625, 4259.16748046875, -108.84466552734375, -8615.59375, 755.1635131835938, 1640.710205078125, 3686.39990234375, 69.50007629394531, -7127.4765625, 2351.5986328125, 5514.26806640625, 10846.193359375, -440.5032958984375, -22264.822265625, -342.37554931640625, -854.77978515625, -1502.192626953125, 146.31192016601562, 3256.03466796875, -3436.03857421875, -8060.07080078125, -15852.783203125, 643.8091430664062, 32544.76953125], "dtype": "float32", "order": "row_major", "shape": [5, 5]}}, "vjp": {"a": {"data": [-347.885986328125, -379.6346740722656, -1217.36767578125, 98.0772705078125, 2011.601806640625, -749.7403564453125, -751.1666259765625, -2515.167236328125, 243.9674072265625, 4076.7734375, -3477.60693359375, -3101.760498046875, -11050.04296875, 1317.23095703125, 17441.701171875, -274.097412109375, -302.6500549316406, -964.537353515625, 75.5528564453125, 1598.3316650390625, 4466.537109375, 4328.01318359375, 14747.810546875, -1524.73095703125, -23723.86328125], "dtype": "float32", "order": "row_major", "shape": [5, 5]}}}}], "provenance": {"fd_policy_version": "v1", "generator": "python-pytorch-v1", "seed": 17, "source_commit": "449b1768410104d3ed79d3bcfe4ba1d65c7f22c0", "source_file": "torch/testing/_internal/common_methods_invocations.py", "source_function": "sample_inputs_matrix_exp", "source_repo": "pytorch", "torch_version": "2.10.0"}, "schema_version": 1} +{"case_id": "matrix_exp_f32_identity_002", "comparison": {"first_order": {"atol": 10000.0, "kind": "allclose", "rtol": 10.0}}, "dtype": "float32", "expected_behavior": "success", "family": "identity", "inputs": {"a": {"data": [-8.478373527526855, -1.765825867652893, -4.3228044509887695, -2.400455951690674, -7.950586795806885, 3.611605167388916, -8.067646980285645, -0.5734938383102417, 3.1285104751586914, -3.033684730529785, 5.106744766235352, 1.135137677192688, 4.947308540344238, 5.774446487426758, -3.973021984100342, 3.270799398422241, -3.893850803375244, 2.8211474418640137, -4.702395439147949, 4.163121223449707, 1.822007417678833, -3.522088050842285, -4.414461135864258, 2.3285136222839355, 8.397199630737305, 4.319086074829102, -0.8690775632858276, -0.43684494495391846, 5.1154985427856445, -6.255067825317383, 2.992039680480957, -2.982185125350952, 5.20722770690918, -3.2104849815368652, 0.44502782821655273, 3.039132595062256, 6.185064315795898, -1.3228504657745361, 8.210625648498535, -7.614414691925049, -1.605404019355774, -8.97458267211914, 0.7455596923828125, 2.5540218353271484, -3.6432604789733887, 3.7379093170166016, -1.4589811563491821, -7.820890426635742, 6.909596920013428, 5.549069881439209, 4.550469398498535, 7.178959369659424, 3.3094801902770996, 4.785210132598877, 7.468499183654785, -1.81334388256073, -7.0191497802734375, -4.426130771636963, -1.2014161348342896, -0.9889798164367676, -0.060997724533081055, 5.157646656036377, 2.8871190547943115, -6.654526233673096, -2.703648567199707, -2.117246627807617, 5.477522373199463, -3.265012264251709, -3.765460968017578, -1.447023868560791, -2.289339542388916, -2.2159457206726074, -8.805970191955566, 8.018885612487793, 4.79000997543335, -4.258679389953613, -5.615394592285156, 0.31310606002807617, 5.127964019775391, -6.4587721824646, -3.3991217613220215, 3.7644824981689453, -5.805131912231445, -1.002150535583496, -6.786881446838379, 8.347665786743164, 4.851363182067871, -8.318924903869629, -4.969023704528809, 3.1903395652770996, 0.49324536323547363, 2.3850836753845215, -7.3627119064331055, -4.818963050842285, 4.083637237548828, -6.862518787384033, -1.8878138065338135, 3.9577925205230713, 4.671151638031006, 0.5595957040786743, 2.6090190410614014, 4.003562927246094, -1.0514216423034668, -2.459080219268799, 6.872914791107178, 8.773387908935547, 4.1688127517700195, -3.9342148303985596, -7.8285956382751465, -8.883143424987793, 0.06226050853729248, -3.4531240463256836, -2.2647950649261475, -1.2656142711639404, 8.51141357421875, 8.530887603759766, -0.8414647579193115, -2.702526092529297, 4.370731353759766, -0.7181721925735474, -8.563591003417969, 2.9337079524993896, 8.616189956665039, -7.134056091308594, -1.9437636137008667], "dtype": "float32", "order": "row_major", "shape": [5, 5, 5]}}, "observable": {"kind": "identity"}, "op": "matrix_exp", "probes": [{"cotangent": {"value": {"data": [-0.06209466495078827, -0.0045469556120126065, -0.06174642109463974, -0.02288351353219217, -0.15641720312610272, -0.052118829448635544, 0.12924054033368929, 0.11731480596788542, -0.002664276350963331, -0.025795343800954263, -0.012248411555092318, -0.22934470659684378, 0.040357823691586726, 0.0006867883371022491, -0.04624047419992971, -0.09390555153852055, 0.05376417465997542, 0.09764442417839435, -0.06759581593708443, 0.07664091002659104, -0.07937093681882113, -0.16557310224826066, 0.2062511556134536, 0.07527237995649728, -2.2781677717850204e-05, 0.08543550069975794, -0.02753247830095784, 0.17561854107310226, -0.009888324709320115, -0.043760999252812335, 0.061336281541291705, 0.0023950156326457985, -0.03660545432520352, 0.07861413665767303, -0.041873294109579226, 0.03139012995722393, -0.06543654831183189, 0.002068093707968985, 0.05143844244071986, -0.024860957191816434, -0.20201800401505884, -0.04265612287357614, 0.1436378930095119, 0.050370669952191284, -0.0008537879177071413, -0.06269752121840205, 0.045852865201902925, 0.097064776468924, 0.07355218248853207, 0.17983878826613447, -0.21797376480408442, -0.05694755437224114, -0.053465264822365154, -0.039402603396828176, -0.051965164951868606, 0.07958044714145131, 0.023569512689841903, 0.02225204331030879, -0.018088016334808128, -0.06399104629497097, -0.01039534416043346, 0.06656682371986775, 0.09796028408655881, -0.09284860474340527, -0.059235523324418074, -0.09127948269549599, -0.10618895847004936, 0.26421263340392426, 0.03358035474430181, -0.029389900559897545, 0.0585276883780753, -0.04567008011139487, -0.18284968704067298, -0.08835699995931638, 0.0689493703404468, 0.010084381146278564, 0.1418894801933319, 0.0075315482495800475, 0.09922472954757917, -0.04650045102932859, -0.03324720576417209, 0.07217699159950405, 0.07093755028651878, 0.08875349749983596, 0.06259126849041381, 0.060856620621639886, 0.00010757684311091888, 0.03338088780413625, -0.10824245020446333, -0.008116003911755676, 0.04610163263299465, 0.03259832726103602, -0.013383131234374575, 0.08842894276427075, 0.06875242914705833, 0.21103359380593048, -0.010631642113797172, 0.05549913545312881, -0.18339763253027166, 0.0698393868802801, -0.010805203385691565, -0.21658131092012275, -0.015563427191662442, 0.02568644611690153, -0.10630225944715595, -0.07512741401242133, 0.004852098791503047, -0.08939023900739719, 0.026514175804060982, 0.04083887041920474, -0.002308729761807498, 0.05414705998998842, 0.10855096384029088, 0.08303125798886953, -0.019763329643416826, -0.10059866128850888, 0.02551516285998381, 0.07583141936040592, 0.06871924426167444, 0.05024861454304022, 0.02983773446245493, -0.14311820012103035, -0.0012477435879278656, -0.0984754917256014, -0.1292230612719222], "dtype": "float32", "order": "row_major", "shape": [5, 5, 5]}}, "direction": {"a": {"data": [0.04978903297351375, 0.020576234552884595, 0.08825036731513794, 0.14170932525955446, -0.2014004255451683, -0.02616117849637527, 0.09270141114238255, -0.03477381109965427, 0.09285424489953766, 0.09761154483694949, 0.01966264251316358, -0.05029277416958518, -0.028053247818474368, 0.11979984285940241, -0.06625863795719156, -0.10734607839662966, -0.15281472837257895, 0.0646055851191745, -0.03135036623181809, 0.09477430418981266, -0.13433435328141932, -0.04792429051792885, 0.04051988502165198, 0.007583651561193618, 0.0385173031021053, 0.08505237701228557, 0.06000985101922711, -0.0837423787013708, -0.06362626592120345, -0.14827358467572596, 0.1942968186088653, 0.006218943172235666, 0.1373262083744433, 0.019700757820196493, 0.02756238122545291, -0.04974873278375838, 0.02210386059400079, -0.04163050579922349, -0.053077149223006125, -0.009529876358752251, 0.11630599000613671, -0.021825402599562407, 0.11471541773593044, 0.0508777602203779, 0.17409042714515832, 0.08921685661354183, 0.04049578611912587, -0.03877376325770975, 0.09242456247332542, 0.15058845022503906, -0.00463421495108455, -0.007815776258342075, 0.006711445633331291, 0.01117313261477549, 0.0019594237336896313, -0.04386065452331688, 0.06356315950463329, -0.03149708306240836, -0.03148251345230175, -0.1368885016730662, -0.010627622533273793, 0.1128159294983296, 0.13038371279019614, -0.12558060668406254, 0.04378671123745495, 0.10650533254479655, 0.07455186417577603, -0.18003286111540945, -0.09823822061075499, 0.07595845181188744, -0.03115540130475743, -0.11081325072123682, -0.06048716500637595, -0.061207043816555055, -0.19991112901527347, -0.0602742721200624, -0.0001116604452250886, -0.06677738462232702, -0.013983740230704282, -0.02067228371099725, -0.05256044774220934, 0.07066292939199488, 0.17298325598634384, 0.0014244831154900692, -0.0887063949938876, 0.07196641589557999, 0.0016468751426552861, -0.01975360292481609, 0.002908925941455456, -0.03767259726675306, 0.18508580008582057, -0.02525019043452808, 0.15984908961400554, 0.048537115662642974, -0.06960775583427167, -0.13518593972783444, -0.12459342712369698, 0.03425869659249733, -0.0030917126619093315, 0.0031970529266971363, 0.09346159386760726, 0.024487167175499668, 0.08232860870568753, 0.11186332063133896, -0.13923692463681048, 0.03168775459247391, 0.05540419274609537, 0.052694960521991824, 0.03395135269758167, 0.04006386106819254, 0.06768665346273105, 0.05240109100185387, -0.005639854721545232, -0.05355187785873231, 0.16370288746029615, 0.1820242075596093, 0.035084336392430075, 0.05936636673588415, -0.1700161070181542, 0.13236844311893975, -0.024698193691405794, -0.17083325688164283, -0.11033196557469815, -0.08596099020160837, -0.036532658455037094], "dtype": "float32", "order": "row_major", "shape": [5, 5, 5]}}, "fd_ref": {"jvp": {"value": {"data": [307.1498718261719, 393.8313293457031, 995.858154296875, 196.27247619628906, -2170.43359375, 75.3405532836914, 42.404266357421875, -38.809478759765625, 27.305273056030273, -254.2834014892578, 602.6000366210938, 659.4767456054688, 1371.1201171875, 342.7889099121094, -3680.45751953125, -107.04266357421875, -116.6861572265625, -238.6378631591797, -60.356536865234375, 650.5364379882812, -785.0943603515625, -800.5399169921875, -1474.5291748046875, -423.27392578125, 4492.19482421875, 305.1702880859375, 2153.424072265625, 2914.665771484375, -1160.236328125, -3616.56005859375, 247.46820068359375, 1691.922607421875, 2323.78759765625, -852.88427734375, -2875.331787109375, 331.0714416503906, 2416.1767578125, 3204.292724609375, -1407.7344970703125, -3986.774169921875, -99.79615783691406, -637.7569580078125, -906.0751342773438, 268.87030029296875, 1113.948486328125, -423.30059814453125, -3019.0166015625, -4063.166748046875, 1656.7568359375, 5040.06005859375, -96.36605834960938, -83.73403930664062, -338.7747802734375, 240.916259765625, 86.40818786621094, 1.244768500328064, 1.9172636270523071, 8.93857479095459, -6.457143783569336, -2.476505994796753, 55.71525573730469, 45.21845626831055, 142.9187469482422, -95.84519958496094, -16.842370986938477, 2.865081787109375, 3.6195037364959717, 40.92000198364258, -33.060569763183594, -23.79470443725586, -93.73284149169922, -72.2396011352539, -230.7939910888672, 155.95970153808594, 30.90325355529785, -4.104122638702393, -2.6132259368896484, 4.718470096588135, 2.3950955867767334, 13.199886322021484, 9.796655654907227, -5.461887836456299, -5.649833679199219, -1.2984896898269653, -17.273805618286133, 0.5970945358276367, -2.3677923679351807, 0.3705631196498871, 0.6427271366119385, 1.1070549488067627, 0.18174932897090912, 2.712822675704956, -1.082129716873169, -0.6362702250480652, -2.618398427963257, -0.1812806874513626, 7.590460300445557, -3.0058155059814453, -2.529007911682129, -7.883730888366699, 170.19082641601562, 190.1879425048828, -121.66801452636719, -191.15606689453125, -126.94438171386719, 1547.26025390625, 1312.777099609375, -815.4967651367188, -674.4202270507812, -773.0213623046875, -1048.966796875, -852.97314453125, 529.642333984375, 366.2327880859375, 491.0727844238281, 459.52618408203125, 365.2707824707031, -208.26991271972656, -139.1524658203125, -187.66432189941406, -820.06689453125, -673.3137817382812, 415.7311096191406, 308.72064208984375, 381.86944580078125], "dtype": "float32", "order": "row_major", "shape": [5, 5, 5]}}, "method": "central_difference", "stencil_order": 2, "step": 0.05494338607788086}, "probe_id": "p0", "pytorch_ref": {"jvp": {"value": {"data": [307.1817626953125, 393.9033508300781, 996.0905151367188, 196.29449462890625, -2170.8408203125, 75.35556030273438, 42.44129180908203, -38.69266891479492, 27.31534767150879, -254.4834442138672, 602.672119140625, 659.6483154296875, 1371.688232421875, 342.84033203125, -3681.44140625, -107.0558090209961, -116.71733856201172, -238.7407684326172, -60.366554260253906, 650.725830078125, -785.2138671875, -800.8026123046875, -1475.337646484375, -423.3511962890625, 4493.68701171875, 305.167724609375, 2153.44873046875, 2914.62890625, -1160.214599609375, -3616.5986328125, 247.46792602539062, 1691.94970703125, 2323.811279296875, -852.8741455078125, -2875.375, 331.06475830078125, 2416.1923828125, 3204.251953125, -1407.705322265625, -3986.79150390625, -99.79818725585938, -637.77294921875, -906.0953369140625, 268.86785888671875, 1113.97705078125, -423.30096435546875, -3019.108642578125, -4063.216064453125, 1656.74169921875, 5040.24658203125, -96.36841583251953, -83.73670959472656, -338.7808532714844, 240.9222412109375, 86.41024780273438, 1.2447164058685303, 1.9172396659851074, 8.938377380371094, -6.457049369812012, -2.476517915725708, 55.7174186706543, 45.21965789794922, 142.92054748535156, -95.84605407714844, -16.841459274291992, 2.8651485443115234, 3.6202125549316406, 40.92311477661133, -33.06300735473633, -23.795970916748047, -93.73536682128906, -72.24061584472656, -230.79812622070312, 155.9625701904297, 30.904315948486328, -4.104260444641113, -2.6130337715148926, 4.718400478363037, 2.3952527046203613, 13.20016098022461, 9.797086715698242, -5.4625468254089355, -5.649773597717285, -1.2986217737197876, -17.274066925048828, 0.596918523311615, -2.367732048034668, 0.3705272674560547, 0.6427620649337769, 1.1073251962661743, 0.18190014362335205, 2.713132858276367, -1.0822882652282715, -0.6365821957588196, -2.6192405223846436, -0.1809748411178589, 7.590392589569092, -3.005769729614258, -2.529266119003296, -7.884451866149902, 170.1984100341797, 190.19732666015625, -121.67578125, -191.16903686523438, -126.95176696777344, 1547.3297119140625, 1312.8485107421875, -815.5444946289062, -674.4818115234375, -773.0632934570312, -1048.9879150390625, -852.9971923828125, 529.6515502929688, 366.251708984375, 491.08251953125, 459.5444641113281, 365.2902526855469, -208.2760009765625, -139.1686553955078, -187.67262268066406, -820.0928955078125, -673.333984375, 415.7474060058594, 308.74542236328125, 381.8801574707031], "dtype": "float32", "order": "row_major", "shape": [5, 5, 5]}}, "vjp": {"a": {"data": [215.5213623046875, 104.01427459716797, 101.65373229980469, -137.40728759765625, -508.4765625, 246.65972900390625, 99.06330871582031, -176.1667022705078, -174.61383056640625, -366.4292907714844, 390.7349548339844, 60.18330383300781, -1684.1663818359375, -360.2438049316406, 452.55426025390625, 28.891624450683594, -9.13512897491455, -321.34619140625, -38.36896896362305, 178.0365753173828, -644.3425903320312, -93.3996353149414, 2865.70654296875, 599.2103881835938, -812.1508178710938, 757.466064453125, 402.856689453125, 860.682861328125, -74.1733627319336, -954.3592529296875, 2007.4228515625, 717.4141845703125, 1788.4962158203125, -77.57438659667969, -2034.816162109375, 3298.0341796875, 1271.5185546875, 3068.8994140625, -161.1156005859375, -3479.59033203125, -469.9233093261719, -125.62106323242188, -357.5208435058594, 4.801294803619385, 414.9906921386719, -4553.3642578125, -1906.39013671875, -4447.42431640625, 276.59716796875, 5020.09716796875, -112.8762435913086, -1.8665539026260376, 74.08976745605469, 10.419020652770996, -124.32998657226562, -106.77130126953125, -0.3381035327911377, 63.273582458496094, 12.803954124450684, -101.00798034667969, -422.0987854003906, 5.609029769897461, 195.1757049560547, 58.440242767333984, -320.27374267578125, 296.7132568359375, -4.758986473083496, -130.06398010253906, -41.604400634765625, 216.85263061523438, 126.36724853515625, -7.1059722900390625, -18.647323608398438, -25.34503936767578, 29.931509017944336, -4.342999458312988, -3.6722960472106934, -2.4750118255615234, 5.806969165802002, 11.310194969177246, 11.181428909301758, -12.323678016662598, 1.9374099969863892, -7.939127445220947, -9.520703315734863, -5.4553141593933105, 10.492610931396484, 0.008542656898498535, 2.2637031078338623, 0.6423772573471069, -2.980515480041504, 6.925246238708496, 0.08112241327762604, 1.279319167137146, -0.7972592115402222, -8.706437110900879, 29.347187042236328, 2.3814308643341064, -0.06823676824569702, -10.390714645385742, -93.60846710205078, -627.642333984375, 370.64398193359375, -204.7105255126953, 379.21563720703125, -83.45327758789062, -565.166015625, 330.3388977050781, -183.07257080078125, 351.97503662109375, 60.9028205871582, 239.88040161132812, -114.11224365234375, 78.13658905029297, -166.35302734375, 62.681907653808594, 398.5475769042969, -223.1473846435547, 126.15220642089844, -264.4777526855469, 51.14750671386719, 229.5498046875, -115.74836730957031, 76.29813385009766, -164.39886474609375], "dtype": "float32", "order": "row_major", "shape": [5, 5, 5]}}}}], "provenance": {"fd_policy_version": "v1", "generator": "python-pytorch-v1", "seed": 18, "source_commit": "449b1768410104d3ed79d3bcfe4ba1d65c7f22c0", "source_file": "torch/testing/_internal/common_methods_invocations.py", "source_function": "sample_inputs_matrix_exp", "source_repo": "pytorch", "torch_version": "2.10.0"}, "schema_version": 1} +{"case_id": "matrix_exp_c64_identity_001", "comparison": {"first_order": {"atol": 10000000.0, "kind": "allclose", "rtol": 10.0}}, "dtype": "complex64", "expected_behavior": "success", "family": "identity", "inputs": {"a": {"data": [[-8.478373527526855, -1.765825867652893], [-4.3228044509887695, -2.400455951690674], [-7.950586795806885, 3.611605167388916], [-8.067646980285645, -0.5734938383102417], [3.1285104751586914, -3.033684730529785], [5.106744766235352, 1.135137677192688], [4.947308540344238, 5.774446487426758], [-3.973021984100342, 3.270799398422241], [-3.893850803375244, 2.8211474418640137], [-4.702395439147949, 4.163121223449707], [1.822007417678833, -3.522088050842285], [-4.414461135864258, 2.3285136222839355], [8.397199630737305, 4.319086074829102], [-0.8690775632858276, -0.43684494495391846], [5.1154985427856445, -6.255067825317383], [2.992039680480957, -2.982185125350952], [5.20722770690918, -3.2104849815368652], [0.44502782821655273, 3.039132595062256], [6.185064315795898, -1.3228504657745361], [8.210625648498535, -7.614414691925049], [-1.605404019355774, -8.97458267211914], [0.7455596923828125, 2.5540218353271484], [-3.6432604789733887, 3.7379093170166016], [-1.4589811563491821, -7.820890426635742], [6.909596920013428, 5.549069881439209]], "dtype": "complex64", "order": "row_major", "shape": [5, 5]}}, "observable": {"kind": "identity"}, "op": "matrix_exp", "probes": [{"cotangent": {"value": {"data": [[-0.03051585802348958, 0.0878156621761248], [0.1672381263425743, -0.047091005874622616], [-0.10836562201792747, 0.2232629890428767], [0.09707478898954414, 0.04142309233296733], [0.056428985905515476, 0.07215016872235298], [-0.09332365273516939, -0.09212979408648944], [0.21771975723862497, 0.029840051267289842], [0.0710320303220959, -0.03509839895476298], [-0.09890400333088453, -0.2501191126694735], [-0.026645223773012087, -0.14512122749004044], [-0.06626462745302795, -0.26387461258940265], [0.01517175094243316, 0.09266052872942501], [0.12643155090977135, -0.049902549437015935], [0.40303401976816655, -0.050471766327624845], [0.2574709579603093, 0.12145067401159666], [-0.01024224113938002, 0.2537009642869875], [0.17803076886313085, 0.04306765894069916], [-0.2808474817345555, 0.007561672260845462], [-0.06487601808374137, 0.03792914266280086], [-0.028727020239884656, 0.14316117350661073], [0.13548060860931854, 0.05367333659714836], [-0.0827558389282218, -0.12675110630235786], [0.11575708949523499, 0.017702796917247823], [0.23646820399694773, 0.19972201605843767], [0.015582465117316698, 0.09489281946943479]], "dtype": "complex64", "order": "row_major", "shape": [5, 5]}}, "direction": {"a": {"data": [[-0.17765898045483589, 0.2089609417431023], [0.3169108128266098, -0.058337101600523085], [0.011470753112386782, 0.07609985997124892], [-0.1016495407324242, 0.0427021806833484], [0.035886266414198065, 0.022608109496864336], [0.042833277378174994, 0.0649865363665304], [0.039809306415280056, 0.2004614681547255], [0.13217385509813478, 0.20917089911062317], [0.0008397617745718083, 0.01991801209905113], [-0.19692831332268834, 0.10399352326086764], [0.059749221359298166, 0.031633761729234856], [0.1624585140811741, 0.0152059285807258], [0.0023223954018387204, -0.09132236516297768], [-0.11877073703928012, 0.1931241957673723], [0.04669479048284154, -0.06986370147277231], [0.0196598084954273, -0.036580664271956366], [0.02579727101242035, -0.028874053554815992], [0.1622294385232679, 0.17024052653886285], [0.18124647982337758, 0.043586065437823326], [0.04480703682035882, 0.08206738032766264], [-0.25986529176305784, 0.08318079512547381], [-0.281353929139904, 0.41547848040729285], [-0.15755305166897604, 0.12877555577894817], [-0.08252738663809935, 0.18573463520382105], [0.1748745493906095, 0.04968037977863982]], "dtype": "complex64", "order": "row_major", "shape": [5, 5]}}, "fd_ref": {"jvp": {"value": {"data": [[9748.439453125, 19917.99609375], [26266.2734375, 12908.423828125], [-31055.861328125, -38384.41015625], [882.507568359375, 22284.19140625], [-35117.61328125, 6985.3505859375], [-4613.31982421875, 124295.8203125], [94348.9296875, 132490.53125], [-50111.484375, -279178.21875], [-51638.36328125, 112802.203125], [-196318.234375, -55574.62890625], [-25597.388671875, -158519.828125], [-152239.546875, -142639.359375], [131488.953125, 345702.46875], [34487.8359375, -156689.453125], [266517.28125, 19452.521484375], [98236.453125, -24901.654296875], [81354.1171875, -104173.2421875], [-207717.296875, 100820.6484375], [101002.7578125, 13961.3623046875], [915.1929931640625, 167975.265625], [39537.73828125, -33565.63671875], [12915.427734375, -65784.1640625], [-76021.71875, 90627.453125], [50405.65625, -17944.3203125], [38678.578125, 80046.703125]], "dtype": "complex64", "order": "row_major", "shape": [5, 5]}}, "method": "central_difference", "stencil_order": 2, "step": 0.0033462886810302736}, "probe_id": "p0", "pytorch_ref": {"jvp": {"value": {"data": [[9731.6513671875, 19909.46484375], [26241.990234375, 12918.5234375], [-31001.2734375, -38371.890625], [870.290771484375, 22271.96484375], [-35096.2421875, 6948.81640625], [-4669.55078125, 124205.875], [94223.671875, 132461.203125], [-49933.875, -278955.6875], [-51659.0625, 112708.921875], [-196115.75, -55667.6640625], [-25516.703125, -158415.765625], [-152070.78125, -142628.59375], [131185.84375, 345462.625], [34530.38671875, -156572.5], [266291.78125, 19600.25390625], [98176.9609375, -24857.400390625], [81344.96875, -104083.390625], [-207581.78125, 100661.21875], [100939.9375, 13983.791015625], [828.177734375, 167845.921875], [39519.890625, -33535.7890625], [12921.8515625, -65744.265625], [-75990.984375, 90537.2578125], [50378.4453125, -17924.443359375], [38623.40625, 80004.515625]], "dtype": "complex64", "order": "row_major", "shape": [5, 5]}}, "vjp": {"a": {"data": [[11886.15234375, -40750.1953125], [-59005.5234375, -172296.375], [33888.36328125, 198335.03125], [141223.65625, 24440.263671875], [52313.22265625, 45524.12109375], [47490.953125, -31185.29296875], [91149.703125, -229122.75], [-140721.375, 234621.046875], [125766.2890625, 147475.0625], [14547.029296875, 91184.5625], [-44350.328125, 99385.5703125], [87735.125, 462880.6875], [-12660.8876953125, -522207.875], [-355924.8125, -110104.5390625], [-118395.578125, -132370.9375], [398.41455078125, -51828.83984375], [-129262.578125, -186455.75], [108861.5546875, 227870.34375], [175653.0, -22935.458984375], [76123.640625, 34534.296875], [-82138.125, -12889.9921875], [-324512.375, 153972.6875], [382896.8125, -108773.8828125], [12444.21875, -286959.40625], [74058.2734375, -114834.5703125]], "dtype": "complex64", "order": "row_major", "shape": [5, 5]}}}}], "provenance": {"fd_policy_version": "v1", "generator": "python-pytorch-v1", "seed": 17, "source_commit": "449b1768410104d3ed79d3bcfe4ba1d65c7f22c0", "source_file": "torch/testing/_internal/common_methods_invocations.py", "source_function": "sample_inputs_matrix_exp", "source_repo": "pytorch", "torch_version": "2.10.0"}, "schema_version": 1} +{"case_id": "matrix_exp_c64_identity_002", "comparison": {"first_order": {"atol": 10000000.0, "kind": "allclose", "rtol": 10.0}}, "dtype": "complex64", "expected_behavior": "success", "family": "identity", "inputs": {"a": {"data": [[-8.478373527526855, -1.765825867652893], [-4.3228044509887695, -2.400455951690674], [-7.950586795806885, 3.611605167388916], [-8.067646980285645, -0.5734938383102417], [3.1285104751586914, -3.033684730529785], [5.106744766235352, 1.135137677192688], [4.947308540344238, 5.774446487426758], [-3.973021984100342, 3.270799398422241], [-3.893850803375244, 2.8211474418640137], [-4.702395439147949, 4.163121223449707], [1.822007417678833, -3.522088050842285], [-4.414461135864258, 2.3285136222839355], [8.397199630737305, 4.319086074829102], [-0.8690775632858276, -0.43684494495391846], [5.1154985427856445, -6.255067825317383], [2.992039680480957, -2.982185125350952], [5.20722770690918, -3.2104849815368652], [0.44502782821655273, 3.039132595062256], [6.185064315795898, -1.3228504657745361], [8.210625648498535, -7.614414691925049], [-1.605404019355774, -8.97458267211914], [0.7455596923828125, 2.5540218353271484], [-3.6432604789733887, 3.7379093170166016], [-1.4589811563491821, -7.820890426635742], [6.909596920013428, 5.549069881439209], [4.550469398498535, 7.178959369659424], [3.3094801902770996, 4.785210132598877], [7.468499183654785, -1.81334388256073], [-7.0191497802734375, -4.426130771636963], [-1.2014161348342896, -0.9889798164367676], [-0.060997724533081055, 5.157646656036377], [2.8871190547943115, -6.654526233673096], [-2.703648567199707, -2.117246627807617], [5.477522373199463, -3.265012264251709], [-3.765460968017578, -1.447023868560791], [-2.289339542388916, -2.2159457206726074], [-8.805970191955566, 8.018885612487793], [4.79000997543335, -4.258679389953613], [-5.615394592285156, 0.31310606002807617], [5.127964019775391, -6.4587721824646], [-3.3991217613220215, 3.7644824981689453], [-5.805131912231445, -1.002150535583496], [-6.786881446838379, 8.347665786743164], [4.851363182067871, -8.318924903869629], [-4.969023704528809, 3.1903395652770996], [0.49324536323547363, 2.3850836753845215], [-7.3627119064331055, -4.818963050842285], [4.083637237548828, -6.862518787384033], [-1.8878138065338135, 3.9577925205230713], [4.671151638031006, 0.5595957040786743], [2.6090190410614014, 4.003562927246094], [-1.0514216423034668, -2.459080219268799], [6.872914791107178, 8.773387908935547], [4.1688127517700195, -3.9342148303985596], [-7.8285956382751465, -8.883143424987793], [0.06226050853729248, -3.4531240463256836], [-2.2647950649261475, -1.2656142711639404], [8.51141357421875, 8.530887603759766], [-0.8414647579193115, -2.702526092529297], [4.370731353759766, -0.7181721925735474], [-8.563591003417969, 2.9337079524993896], [8.616189956665039, -7.134056091308594], [-1.9437636137008667, 1.9517877101898193], [-7.32878303527832, -7.705445766448975], [6.718801498413086, -3.186859130859375], [5.249946594238281, 0.8045790195465088], [2.1958847045898438, -5.998164176940918], [-1.922160029411316, 5.936859130859375], [4.127967834472656, -2.602128028869629], [1.1071697473526, 6.89056396484375], [-7.885776042938232, 5.781133651733398], [-2.7048606872558594, -1.83092200756073], [4.28273868560791, -7.476827144622803], [-1.3596267700195312, 8.600601196289062], [3.239168643951416, -3.327920913696289], [-1.960387945175171, 7.097623825073242], [3.4002695083618164, 6.100860595703125], [-5.795559406280518, 2.5947282314300537], [1.608823299407959, 1.5688358545303345], [8.356966018676758, 4.982141971588135], [6.203725337982178, 3.951810121536255], [3.7825584411621094, -1.565478801727295], [-7.943257808685303, -7.602569103240967], [-5.605430603027344, -5.826987266540527], [4.236800193786621, 2.786085605621338], [-2.56626033782959, -6.855833530426025], [8.860182762145996, 6.7630720138549805], [4.280237197875977, 8.908514022827148], [-4.791670799255371, -5.125319480895996], [-5.26807975769043, -0.4357280731201172], [-7.944648265838623, 7.124773025512695], [0.2315218448638916, 4.48195743560791], [-4.942539691925049, -0.927196741104126], [1.1837446689605713, -2.463348388671875], [8.495100021362305, -4.1118693351745605], [2.794466972351074, -6.1065778732299805], [-1.0652693510055542, -2.3876960277557373], [3.883119821548462, 6.286882400512695], [2.1888680458068848, -0.8171553611755371], [-2.304619312286377, 7.056262969970703], [-2.1265969276428223, 6.497251033782959], [-4.004391193389893, 7.417506217956543], [8.963968276977539, 6.169381141662598], [5.161905765533447, 3.3498120307922363], [-4.425276756286621, 0.10741281509399414], [3.0989856719970703, -2.3516921997070312], [1.6134077310562134, 5.271086692810059], [-8.338794708251953, -8.19182014465332], [-5.3852715492248535, 1.0978699922561646], [-1.668221354484558, 1.9394280910491943], [-2.1402130126953125, -5.8917412757873535], [1.1949294805526733, 6.095582962036133], [-5.403924942016602, 7.8891825675964355], [2.215019702911377, 8.874296188354492], [-8.249293327331543, -8.614898681640625], [3.0533761978149414, -8.396015167236328], [-7.375467300415039, 4.576351642608643], [4.368868827819824, -8.200986862182617], [6.4762067794799805, -1.7500169277191162], [6.681774139404297, -5.281617164611816], [-8.55782699584961, -5.9662370681762695], [-1.3952507972717285, -8.988519668579102], [-0.6132087707519531, -4.491751670837402], [-2.0947675704956055, -0.0012799501419067383], [0.5051232576370239, -8.80363941192627]], "dtype": "complex64", "order": "row_major", "shape": [5, 5, 5]}}, "observable": {"kind": "identity"}, "op": "matrix_exp", "probes": [{"cotangent": {"value": {"data": [[0.032709822262245995, -0.0018491731140361145], [0.0433689917342817, 0.08694370890128598], [0.06650374126099422, -0.015829404914545146], [-0.08057432671862574, 0.020436324306357192], [0.060737040271860214, 0.05504055727999892], [0.040246539068355326, 0.023898482471424155], [-0.11463027518801792, -0.0009993780466525208], [-0.07887377639027816, -0.10350097780268079], [-0.013167933642857968, -0.07032102390184307], [0.007072914901435358, -0.03765701153468236], [0.010749104499658307, -0.0038156882165911797], [-0.02665079774157242, 0.03641456015718759], [-0.07837390948433903, 0.03079767171833858], [-0.08016530474247029, -0.07976203461457146], [-0.022831753038401435, -0.02811974677062327], [-0.07844071884096744, -0.022211510826298964], [-0.1415877649059005, 0.01648502807147745], [0.11592254859827439, 0.07807330090712412], [0.012486057433706444, -0.030745854792609644], [-0.005419031750608318, -0.06833592099756686], [-0.016941433877879088, 0.008415496960501782], [-0.04529763413844304, -0.03189485912503108], [0.02818958292563801, 0.06388761887971502], [0.03379239535500667, 0.0114867138459894], [-0.1694275151306585, -0.05662687158063323], [-0.027301388031280176, -0.09563838004981451], [-0.014541970656416502, -0.07749918151449396], [0.00906879553627338, 0.023977186680049573], [0.01438594711584274, -0.11644206268536987], [0.022119511056515592, -0.16133840482507486], [-0.025849497106612142, -0.032696996087668566], [0.10026945981200759, -0.05253750491056232], [0.030505227114863556, 0.07443701522184278], [-0.01445105680876189, 0.05064146997258635], [-0.07368428305945096, 0.02990905027823892], [-0.009603421263558617, -0.03585011161823241], [0.011479194347475156, 0.082604148007923], [-0.11484191638177128, -0.001725362042458303], [0.009658197001125522, 0.046190630932025474], [-0.010040972444319821, -0.06928089304171689], [-0.0032742780371382884, 0.016421895576396516], [0.06300292203296921, 0.007611116441768087], [0.09607702553460769, -0.06701144617777008], [-0.014508624651424867, -0.006042151283822969], [-0.03409741467596242, 0.0006421098409135164], [-0.06217123979279075, -0.03253018131235557], [0.09027393963490217, -0.06397185514446808], [0.06369817296569803, -0.05776497757621869], [-0.031940289040503556, -0.022005792844157022], [0.0654397983451321, 0.044612027982356965], [-0.00034478810481169156, -0.08844078387226527], [0.0073051890812964445, 0.05891121588057631], [-0.012698003169943598, -0.006092396602697461], [-0.030953360914105974, -0.021935057031529606], [-0.0720475693060551, -0.035223357598647705], [0.039700120934556916, 0.016237344693862166], [0.08060788413384314, -0.0019782842260037256], [0.0009071794394982305, -0.14507319123431647], [-0.06829667878931864, -0.09856677116701537], [-0.0075554191605078655, -0.06464060436184871], [0.05608756130076852, 0.032732673193078966], [0.05001385168564702, -0.003731962549442828], [-0.017272805902570153, 0.00332948893518089], [0.023889307081361943, -0.03855729499613393], [0.050102457715947374, -0.006517261772929589], [0.04426539716850844, 0.0378854649636573], [-0.017916981242344036, 0.053559903388757456], [-0.13751506891001897, 0.06521151255422161], [-0.05535607446908106, 0.03859875375212353], [0.00824673385835057, 0.05935845933543186], [-0.10692200876259679, -0.048606232111161427], [-0.10406661098395825, 0.03715255507106697], [0.154637620815401, 0.07357172713767275], [-0.07816236514813404, -0.12520071942046299], [0.034911278852532444, 0.03405135518642558], [0.00933090603198182, 0.03444872817974525], [-0.028855256915858114, 0.03459969929532028], [-0.19066584228920164, 0.0013387008886278817], [-0.0666021783324534, -0.038010593740271076], [0.030210487732550396, -0.0842965772033694], [-0.04795961111829138, -0.019967058797765235], [0.013748035851744701, -0.04157498132234465], [0.05014444546316265, -0.03707924880811762], [0.015954569081416976, 0.04933530868001079], [-0.04073958124242519, 0.08855345900333378], [0.024461088854315277, 0.02774728851897477], [0.004859776266361517, 0.025145163916079155], [-0.03466431445594923, 0.04299484592611749], [0.0771602546010306, 0.0018848570879505498], [0.009120433648066877, 0.035700075550528114], [-0.08734817112208774, -0.04290687692046214], [-0.00010933928638058514, 0.08995807971911043], [-0.04224913221069166, -0.09959324511282239], [0.03545776404155664, 0.008275100081504744], [0.022052921492016167, 0.04820039153294066], [0.03853081935781799, 0.08058074911914026], [-0.06347303131981902, 0.02898284624672933], [0.08557792061182215, -0.01966678363403411], [-0.06842483622696391, -0.010387864959813444], [-0.05504113469999877, 0.10386845534116926], [0.045013238024707934, -0.052180238392845496], [-0.09501764237409925, 0.1597391227888309], [0.021509258210410616, -0.0509889650537895], [0.06723034423707001, 0.0012819194314687388], [-0.07608632790710645, -0.017252694922768746], [-0.06794205350243986, -0.04458703128429872], [-0.050657481270388785, 0.08330701343406925], [0.20934499930490286, 0.0830509369773593], [0.11040900716352849, -0.012498961839380627], [0.010049634675640228, -0.010672450132032375], [-0.014130303724132676, -0.06050333790882207], [0.05912102795678174, -0.0665580187409806], [0.0474931377147955, -0.006544801447293648], [-0.05822432077403895, 0.11505064439849311], [0.08445508085597479, -0.1355846681688258], [-0.036235843079296934, -0.1444848933867221], [-0.03431639841693974, -0.012551173645883658], [0.021562233701429585, 0.06599023979247788], [-0.06248408222342195, -0.06096813865696353], [-0.037524644518973624, 0.016506711123794593], [0.010480917124112681, 0.05848603359500555], [-0.030450506325328894, -0.04552241443116085], [-0.06509834568483065, -0.18212619537211808], [-0.10330653254905216, -0.07457652734438842], [0.03502240053753682, 0.06549768935673048]], "dtype": "complex64", "order": "row_major", "shape": [5, 5, 5]}}, "direction": {"a": {"data": [[0.032821989551538325, 0.013564290997695523], [0.05817651944287531, 0.09341779812234777], [-0.13276743485145887, -0.01724600366961789], [0.061110744257237164, -0.022923634621830908], [0.06121149100500534, 0.06434760047037298], [0.012962031715359399, -0.033154065643936005], [-0.01849329601619645, 0.07897460220522892], [-0.04367909820182089, -0.07076482259627646], [-0.10073872349311482, 0.04258938004214043], [-0.020666827750613855, 0.062477236576516215], [-0.08855605601210802, -0.03159270712036005], [0.02671157114941622, 0.004999304038263028], [0.025391427027976164, 0.05606833674888644], [0.039559769172586884, -0.055204754879317336], [-0.041943790978337864, -0.09774509520145998], [0.12808459607707248, 0.0040996594970562255], [0.09052835129601185, 0.012987157866335933], [0.01816970799748572, -0.032795420781937576], [0.014571335654941241, -0.027443715468026497], [-0.03498958689869073, -0.0062822972376700426], [0.07667138194359463, -0.014387769187098557], [0.07562283941633104, 0.0335397002333261], [0.1147641132148043, 0.058813648372667324], [0.0266956827867764, -0.025560490009805373], [0.060928238565774714, 0.09927111562233926], [-0.0030549731107955666, -0.005152325920554825], [0.004424327384259386, 0.007365566920039947], [0.001291693847465154, -0.028913877025224], [0.04190218321225952, -0.020763545597046747], [-0.02075394179894934, -0.09023980521942808], [-0.007005954413915921, 0.07437065762638442], [0.08595171309422477, -0.08278540270132581], [0.02886513160315084, 0.07021058137310485], [0.04914617761541228, -0.11868148681251899], [-0.06476072770132976, 0.0500734284195364], [-0.020538303376579838, -0.07305045203765631], [-0.03987442580807863, -0.040348986799479306], [-0.13178567187605492, -0.039734082950932115], [-7.360893281568165e-05, -0.044021068664950425], [-0.00921837864352794, -0.013627609754973133], [-0.03464896499060823, 0.046582503690480705], [0.11403424691114225, 0.0009390495548401775], [-0.05847714291023946, 0.047441794025756476], [0.0010856551486569247, -0.013021993986180154], [0.0019176259519443175, -0.02483457810167491], [0.12201249228450502, -0.016645462677379683], [0.10537592001215228, 0.03199669993978275], [-0.04588691005652238, -0.0891174497924352], [-0.08213463175887564, 0.02258405952029045], [-0.002038122558459892, 0.0021075652271406735], [0.06161187029294841, 0.016142460958060205], [0.05427276550607765, 0.07374267800928722], [-0.09178794641342698, 0.020889240613185114], [0.03652362041875834, 0.03473763807288622], [0.022381454025633823, 0.04309142994981759], [-0.11482412018910727, -0.085591492434478], [0.0558813011045588, -0.10467879900921803], [0.061186632143599595, -0.052149882817018045], [0.0007496448005777364, -0.0029077519709833763], [0.08775467903805115, -0.015545179594734036], [0.12706011147382482, 0.10743625135371243], [-0.022524885148343863, 0.014453046515691027], [-0.005810517641053881, 0.021644632605196047], [-0.08967455203846904, -0.06672115421070358], [-0.0037179122479743688, -0.035302537351277655], [0.10791641890684181, 0.11999423437029637], [0.023128339317506883, -0.00439249757359461], [0.10194680529784919, -0.006321493809676025], [-0.01628157446944561, -0.11261691555670597], [-0.07273317652316437, -0.05666731125660178], [-0.024083105114938795, 0.03422671452398794], [0.03629422077444476, 0.03800733271900446], [-0.017348057994241068, -0.11858034518399119], [-0.03951143725772428, 0.0979776278148193], [0.08893669871264996, 0.017515716541670463], [-0.0019462684276097123, -0.04385456682012353], [-0.17386689253461446, 0.030595383607852574], [0.0005206562187530314, -0.03505504024724445], [-0.07119007192149583, 0.06835923881093363], [-0.0949744329779689, 0.05842208312130241], [0.05810169698950585, -0.06017133783123878], [-0.12552144741805701, 0.15635959328611743], [0.057064214848802065, 0.2548327070473076], [-0.02900887565023643, 0.04042532917124433], [0.13313692403318636, -0.007496368377699656], [-0.0331753407761836, 0.04649921365252031], [0.0018156683684506399, -0.05211048042272933], [0.0037887942756504515, -0.06460898918150676], [0.023796948812008813, -0.04960763676123166], [0.0015678278392284822, 0.03899563357868341], [-0.018847163193532374, -0.07131197831751586], [0.12144116688054414, 0.04923968239906045], [0.038186150373183576, -0.0006472590997510583], [-0.047531171187879466, 0.03476119308272715], [0.07358509078387349, 0.06260419446602736], [-0.05587951296526991, -0.006278076483890227], [-0.043172104834067984, -0.040532174219045654], [-0.02987123003668952, -0.03939494370834332], [0.06033016930357438, 0.017991374631030673], [-0.0013347025554837646, -0.00911160065094995], [-0.04851179794977693, -0.00788074055937825], [0.05046450193259688, 0.07426398021664143], [-0.07038879925556416, -0.005733477708701686], [-0.011730963007537327, -0.0706264057141557], [0.200300359595836, 0.025457364664253957], [-0.022280567579483705, 0.04437000914601644], [-0.0346226197384186, -0.05476226118583074], [-0.048577009154468616, -0.05685206702333609], [0.007644999074185666, 0.10756683022528359], [0.005709688936901634, 0.07522255698593279], [-0.03525212672449172, -0.07406686766236287], [0.06709238682823776, 0.02156578683842658], [0.06728428397625706, 0.04745062296349537], [0.04613558435225411, 8.15543219445183e-05], [0.02530614768505688, -0.07271211373245733], [-0.1681553074250086, 0.023944304527928943], [0.02471288777248562, -0.010145790566452574], [0.07189539601341786, 0.07563873150426104], [0.010397588518004581, 0.014255441291785468], [0.07180074384039266, -0.05741086055111401], [-0.04140707352019735, 0.0025875181104231333], [0.03226370638851578, 0.06005696769726338], [-0.01594463314806608, 0.007463259861178814], [0.008655142707204185, 0.021260031783832338], [0.1407914565906172, 0.0042041692535727095]], "dtype": "complex64", "order": "row_major", "shape": [5, 5, 5]}}, "fd_ref": {"jvp": {"value": {"data": [[-19319.484375, -4860.60888671875], [-19651.068359375, 8019.974609375], [47982.37109375, 6566.330078125], [-17023.083984375, -9156.9052734375], [15453.505859375, -33375.25390625], [-47122.67578125, -77180.15625], [-87770.5078125, -39534.484375], [134593.0625, 172994.84375], [-24435.556640625, -83131.234375], [154768.859375, -58533.03515625], [68012.9453125, 77625.9765625], [105963.578125, 27266.365234375], [-185144.84375, -168349.765625], [43712.92578125, 88511.125], [-162566.15625, 96127.390625], [-41108.84375, 46462.41015625], [-9112.0234375, 64477.34375], [85992.5390625, -123308.5625], [-49068.12890625, 32310.3125], [-69682.0, -89077.484375], [-2503.57958984375, 30784.05859375], [17046.802734375, 28830.375], [-2170.54345703125, -74820.828125], [-10100.482421875, 28262.568359375], [-54638.25390625, -15284.80859375], [-363055.84375, 114781.2890625], [939115.375, -159547.53125], [-1801818.375, 101800.625], [1436547.5, 130798.0], [-1185256.25, -186190.421875], [376113.96875, -331634.28125], [-1007318.125, 662385.625], [2032852.5, -1004065.1875], [-1756819.75, 553862.3125], [1466815.0, -358625.6875], [-448375.0625, 584772.6875], [1280841.5, -1283753.625], [-2672220.0, 2128519.0], [2382494.0, -1361018.0], [-2036012.5, 1001772.6875], [179560.6875, -618417.125], [-612352.75, 1463775.125], [1443796.75, -2602824.5], [-1472174.625, 1872313.625], [1302837.25, -1479362.0], [-299964.875, 223877.71875], [783300.9375, -444068.8125], [-1554829.5, 651015.9375], [1327063.75, -320148.40625], [-1102024.25, 197157.390625], [106846.5078125, 913.9130859375], [19759.498046875, 36757.89453125], [-79680.828125, 117043.3203125], [167207.375, 5801.0244140625], [-84134.21875, 38787.08984375], [-33241.65234375, -192944.09375], [54389.4453125, -43256.9375], [226317.40625, 98113.9921875], [-46663.78125, -283638.34375], [88541.140625, 133067.296875], [-218099.9375, -105642.15625], [-3485.26171875, -88343.640625], [266909.09375, -157580.296875], [-324079.0, -163962.203125], [201606.90625, 6885.328125], [71021.890625, -153183.0625], [61538.10546875, -1818.1448974609375], [109847.703125, 184927.984375], [113619.53125, -223933.875], [-4840.64111328125, 141335.109375], [-218518.171875, 94411.8671875], [-70185.2421875, -53835.67578125], [52604.21875, -303326.1875], [-333934.53125, 140169.484375], [133121.515625, -148689.015625], [-1182.4476318359375, -264.7149963378906], [-2252.83447265625, 2357.150390625], [823.0830078125, 2246.90478515625], [2922.174072265625, -1074.002197265625], [1188.8614501953125, -1581.104736328125], [-10621.3349609375, -6779.318359375], [-20268.732421875, 16876.654296875], [7458.4599609375, 21131.3359375], [24779.2421875, -7081.5859375], [10935.9765625, -15397.595703125], [6583.75634765625, -7033.45166015625], [-9717.8291015625, -16747.166015625], [-16508.5234375, 3257.0859375], [2363.92724609375, 18686.328125], [10405.9267578125, 9656.111328125], [1864.2216796875, 1668.3270263671875], [4296.43212890625, -3202.023681640625], [-1122.8927001953125, -4648.8310546875], [-5287.0693359375, 1179.966796875], [-2748.39501953125, 3081.128662109375], [1187.9425048828125, 3886.9140625], [8068.865234375, 1057.196044921875], [2784.4580078125, -6003.96826171875], [-6722.4404296875, -3913.174560546875], [-5403.24462890625, 1073.7449951171875], [4029.349853515625, 549.8968505859375], [-92847.375, 41476.55078125], [63931.31640625, -1483.1724853515625], [84354.953125, 14827.8525390625], [-23570.05078125, -38771.6015625], [-5737.6123046875, 535.3749389648438], [-8292.986328125, -7280.32470703125], [-11169.2958984375, 5832.0048828125], [-3349.23974609375, -4230.720703125], [10709.4345703125, -8339.45703125], [3051.7822265625, -4364.322265625], [-55617.03125, 35615.50390625], [42891.953125, -13879.0869140625], [63416.2734375, 91.00405883789062], [-16659.55078125, -19730.453125], [3059.285400390625, -4204.5810546875], [87948.640625, 129001.15625], [-6854.34814453125, -94225.984375], [2290.591552734375, -119831.046875], [-62726.96484375, 48388.98828125], [14.037859916687012, 4386.58935546875], [44061.453125, -13120.07421875], [-25155.845703125, 7288.0693359375], [-45289.0390625, -3780.540771484375], [2522.942138671875, 14796.8720703125]], "dtype": "complex64", "order": "row_major", "shape": [5, 5, 5]}}, "method": "central_difference", "stencil_order": 2, "step": 0.008069695281982422}, "probe_id": "p0", "pytorch_ref": {"jvp": {"value": {"data": [[-19323.759765625, -4861.689453125], [-19653.857421875, 8022.052734375], [47992.99609375, 6563.66455078125], [-17025.0546875, -9159.333984375], [15458.4716796875, -33383.3984375], [-47128.328125, -77199.2578125], [-87784.640625, -39543.50390625], [134624.625, 173017.78125], [-24433.099609375, -83147.6171875], [154812.15625, -58530.953125], [68032.59375, 77645.6875], [105985.5703125, 27265.515625], [-185205.90625, -168367.390625], [43719.48046875, 88532.1875], [-162620.734375, 96158.265625], [-41124.54296875, 46472.328125], [-9115.7724609375, 64491.7421875], [86011.3671875, -123333.0625], [-49085.16015625, 32311.375], [-69688.15625, -89115.8671875], [-2509.5927734375, 30788.833984375], [17043.828125, 28837.80078125], [-2161.6865234375, -74829.9609375], [-10109.4326171875, 28263.48046875], [-54644.6953125, -15300.7314453125], [-363319.625, 115018.171875], [939955.5, -159893.359375], [-1803489.5, 101973.125], [1437950.25, 130742.578125], [-1186245.5, -186232.796875], [376225.875, -332113.1875], [-1007954.0, 663373.375], [2034490.5, -1005490.0], [-1758169.0, 555084.75], [1468003.0, -359318.1875], [-448489.25, 585549.75], [1281647.0, -1285505.75], [-2674533.25, 2131190.0], [2384326.5, -1363274.125], [-2037749.0, 1003166.375], [179373.59375, -619086.625], [-612407.3125, 1465513.0], [1444808.75, -2605878.25], [-1473242.375, 1874845.875], [1303899.125, -1481082.5], [-300044.34375, 224191.390625], [783724.0, -444749.8125], [-1555887.0, 651968.625], [1327988.0, -320932.6875], [-1102784.375, 197586.53125], [106757.1015625, 1058.55810546875], [19715.25, 36755.8125], [-79740.109375, 116857.046875], [167087.109375, 5966.21923828125], [-84099.9296875, 38669.0625], [-33028.84375, -192836.5], [54388.84765625, -43197.234375], [226072.609375, 98263.3984375], [-46413.84765625, -283507.15625], [88376.5390625, 133032.703125], [-217799.921875, -105771.359375], [-3415.5888671875, -88291.171875], [266815.375, -157183.53125], [-323709.3125, -164110.625], [201425.234375, 7053.423828125], [71102.234375, -152959.796875], [61493.42578125, -1773.2353515625], [109585.921875, 184835.671875], [113695.3203125, -223655.0625], [-4939.55029296875, 141185.09375], [-218442.40625, 94042.921875], [-70087.453125, -53859.6640625], [52909.51953125, -302986.8125], [-333789.75, 139716.53125], [133141.453125, -148406.359375], [-1182.3468017578125, -264.3578186035156], [-2252.45361328125, 2356.98779296875], [823.0245361328125, 2246.4833984375], [2921.840087890625, -1074.092041015625], [1188.6795654296875, -1580.773681640625], [-10620.84765625, -6778.2177734375], [-20267.150390625, 16876.046875], [7458.53564453125, 21131.1953125], [24777.689453125, -7081.7001953125], [10935.8330078125, -15396.748046875], [6582.8447265625, -7032.6416015625], [-9716.44921875, -16745.29296875], [-16507.3125, 3256.87158203125], [2363.76513671875, 18684.10546875], [10404.1357421875, 9655.5634765625], [1863.66845703125, 1668.119384765625], [4296.08056640625, -3201.06201171875], [-1122.2418212890625, -4648.373046875], [-5286.39697265625, 1179.2509765625], [-2748.40771484375, 3080.219482421875], [1187.87548828125, 3886.6337890625], [8068.0703125, 1057.5732421875], [2784.25927734375, -6003.2080078125], [-6721.736328125, -3913.3349609375], [-5402.74560546875, 1073.1361083984375], [4030.369873046875, 514.638671875], [-92761.796875, 41460.92578125], [63887.5703125, -1528.44921875], [84312.3203125, 14794.51953125], [-23551.25, -38715.328125], [-5750.11328125, 540.19091796875], [-8316.32421875, -7318.287109375], [-11178.1572265625, 5857.07568359375], [-3355.019775390625, -4205.345703125], [10724.6171875, -8355.919921875], [3051.30615234375, -4388.03515625], [-55560.3828125, 35631.765625], [42879.5390625, -13910.841796875], [63388.83203125, 61.8515625], [-16659.77734375, -19711.984375], [3031.8916015625, -4204.59130859375], [87968.609375, 128941.1171875], [-6899.5703125, -94210.3359375], [2250.24609375, -119803.828125], [-62695.0, 48388.875], [21.58935546875, 4410.7822265625], [44003.27734375, -13079.734375], [-25112.0234375, 7309.115234375], [-45246.0078125, -3765.94140625], [2492.1611328125, 14772.5302734375]], "dtype": "complex64", "order": "row_major", "shape": [5, 5, 5]}}, "vjp": {"a": {"data": [[-6529.49365234375, 5169.365234375], [2131.312255859375, 35705.2109375], [9773.71875, -36503.21875], [-23926.982421875, -11993.765625], [-8692.1943359375, -9193.966796875], [-8986.927734375, 411.92816162109375], [-18964.62109375, 34537.3359375], [29672.0625, -27011.42578125], [-14222.8359375, -24898.44921875], [-2530.22998046875, -14030.2080078125], [17899.41015625, -11010.646484375], [6926.572265625, -89401.609375], [-36945.01953125, 88401.6328125], [56102.453125, 37862.11328125], [18453.81640625, 25521.30859375], [-3969.8955078125, 10524.2890625], [26978.75390625, 46115.8046875], [-15261.787109375, -54498.30859375], [-38748.3046875, 2594.13134765625], [-16815.4296875, -6709.8349609375], [11257.9296875, 12622.203125], [68590.3828125, -4648.55126953125], [-73115.546875, -19689.119140625], [-23134.4140625, 48408.3125], [-16729.3984375, 17152.009765625], [261897.8125, 15026.140625], [-205063.40625, -194075.15625], [370083.53125, 328776.40625], [-157267.625, -406905.96875], [223684.09375, 179857.40625], [-637158.375, 206018.890625], [670153.5625, 258417.546875], [-1184559.0, -448712.15625], [761294.5625, 824385.0], [-677572.1875, -218031.421875], [976052.9375, -633770.0], [-1222123.75, -138063.125], [2198544.25, 210714.703125], [-1642301.625, -1024809.25], [1247104.5, 86059.4375], [-722910.375, 543727.25], [952573.0625, 55392.1796875], [-1715056.25, -38988.85546875], [1317326.25, 706329.75], [-990899.5, -2742.609375], [597988.375, -479652.9375], [-820520.125, -19906.19140625], [1448542.75, -8837.166015625], [-1137454.75, -573031.5], [835768.3125, -40343.4140625], [256027.109375, 5498.35205078125], [-11863.2734375, 289980.84375], [-381511.625, 227068.21875], [203088.359375, 234230.625], [-413482.03125, -172973.6875], [45552.359375, -47229.78515625], [50548.08203125, 52338.5625], [-24994.447265625, 110730.5], [77215.34375, 2870.3154296875], [-104090.7578125, 46099.4765625], [-143182.96875, -259838.4375], [298837.03125, -155822.46875], [450130.4375, 252245.96875], [118242.71875, -339407.3125], [66315.4609375, 516202.09375], [349511.0, 28436.576171875], [-41732.5234375, 398197.96875], [-543010.8125, 280722.9375], [257714.5625, 340187.75], [-553678.125, -271736.875], [-195381.5625, -60166.875], [75092.0, -223455.140625], [345291.6875, -89918.46875], [-103771.765625, -228861.015625], [283079.4375, 226245.28125], [-371.77105712890625, 184.37109375], [558.1585693359375, -788.8945922851562], [-794.347412109375, -1108.6571044921875], [-81.21290588378906, 266.00555419921875], [-238.857666015625, 135.28933715820312], [-558.18310546875, 343.68841552734375], [914.1826782226562, 1160.26123046875], [894.9468994140625, -1217.9910888671875], [98.78849792480469, -187.27734375], [-1017.9554443359375, -118.59664916992188], [-367.1058654785156, 98.70967102050781], [-1419.0009765625, 1559.2083740234375], [1251.240966796875, 1219.0546875], [523.2318725585938, -208.91799926757812], [-398.57159423828125, -742.0397338867188], [317.3942565917969, -714.3040161132812], [-1490.326171875, -1038.745849609375], [-507.0342102050781, 1595.601806640625], [-27.888214111328125, 346.4730224609375], [1156.4937744140625, -422.49029541015625], [198.53497314453125, -333.8431091308594], [137.44859313964844, -1713.0263671875], [-1513.7344970703125, -128.56320190429688], [-224.801513671875, 493.374755859375], [628.3302612304688, 182.4438018798828], [-60896.44921875, 82038.015625], [-13867.4619140625, -33941.55859375], [-65966.5234375, 55628.859375], [-78852.375, -69071.0546875], [60031.89453125, -39614.703125], [256664.8125, 196591.890625], [-104697.703125, 17580.86328125], [163256.84375, 210353.625], [-214103.609375, 292814.0], [-103699.890625, -181464.78125], [-196976.71875, 79932.0546875], [20392.021484375, -64354.3515625], [-173716.46875, 24567.794921875], [-87111.6875, -223955.359375], [138260.421875, -2468.181640625], [-164839.78125, 73544.421875], [13942.0439453125, -49913.46875], [-143779.984375, 26947.15234375], [-82009.296875, -197199.09375], [112980.7734375, -9019.853515625], [124391.3359375, -153440.171875], [8586.71484375, 64801.875], [131036.828125, -95839.8125], [175692.53125, 133830.3125], [-112732.0546875, 59629.3046875]], "dtype": "complex64", "order": "row_major", "shape": [5, 5, 5]}}}}], "provenance": {"fd_policy_version": "v1", "generator": "python-pytorch-v1", "seed": 18, "source_commit": "449b1768410104d3ed79d3bcfe4ba1d65c7f22c0", "source_file": "torch/testing/_internal/common_methods_invocations.py", "source_function": "sample_inputs_matrix_exp", "source_repo": "pytorch", "torch_version": "2.10.0"}, "schema_version": 1} diff --git a/docs/generated/pytorch-upstream-publish-coverage.md b/docs/generated/pytorch-upstream-publish-coverage.md index 7e54680..7944053 100644 --- a/docs/generated/pytorch-upstream-publish-coverage.md +++ b/docs/generated/pytorch-upstream-publish-coverage.md @@ -9,7 +9,7 @@ and the checked-in `cases/` tree. - AD-relevant scalar upstream variants: 138 - Mapped publishable success families: 174 - Explicit publishable error families: 2 -- Total tracked DB families: 176 +- Total tracked DB families: 177 ## Publishable Family Coverage @@ -93,6 +93,7 @@ publishable upstream coverage that is not yet materialized in this repository. | lu_factor | identity | success | float64, complex128, float32, complex64 | float64, complex128, float32, complex64 | - | | lu_factor_ex | identity | success | float64, complex128, float32, complex64 | float64, complex128, float32, complex64 | - | | lu_solve | identity | success | float64, complex128, float32, complex64 | float64, complex128, float32, complex64 | - | +| matrix_exp | identity | success | float64, complex128, float32, complex64 | float64, complex128, float32, complex64 | - | | matrix_norm | identity | success | float64, complex128, float32, complex64 | float64, complex128, float32, complex64 | - | | matrix_power | identity | success | float64, complex128, float32, complex64 | float64, complex128, float32, complex64 | - | | max_binary | identity | success | float64, float32 | float64, float32 | - | diff --git a/docs/math/eig.md b/docs/math/eig.md index a42485c..06e1f46 100644 --- a/docs/math/eig.md +++ b/docs/math/eig.md @@ -167,3 +167,37 @@ column-wise phase ambiguity of raw eigenvectors. ### `eigvals/identity` The eigenvalue-only family reuses the diagonal part of the same differential. + +## Complex Oracle Strategy + +### Phase ambiguity resolution + +The `values_vectors_abs` observable publishes eigenvalues together with +`abs(eigenvectors)`. Raw eigenvectors are defined only up to per-column complex +phase $V \mapsto V \operatorname{diag}(e^{i\phi_k})$. Taking the element-wise +absolute value collapses this gauge freedom into a well-defined observable whose +AD is unambiguous. + +### Real-to-complex output handling + +For real input matrices ($A \in \mathbb{R}^{N \times N}$), PyTorch's +`linalg.eig` returns complex-valued eigenvalues and eigenvectors. The forward +rule operates in the complex domain; the reverse rule projects the cotangent +back to the real domain via $\bar{A} \leftarrow \operatorname{Re}(\bar{A})$ +(`handle_r_to_c`). The oracle DB includes float32 and float64 cases that +exercise this path. + +### Complex-input coverage + +The oracle DB includes complex64 and complex128 input cases. For complex inputs +the full complex formula applies: the normalization correction uses +$\operatorname{Re}(V^\dagger \dot{V}_{\mathrm{raw}})$, the reverse rule uses +$V^{-\dagger} G V^\dagger$, and the gauge invariance check verifies +$\operatorname{Im}(\operatorname{diag}(V^\dagger \bar{V})) = 0$. + +### Future considerations + +If downstream `tenferro-rs` requires a different observable representation +(sorted eigenvalues, a different gauge-fixing convention, or separate real/imaginary +parts), a new DB family can be added alongside `values_vectors_abs` without +breaking the existing contract. diff --git a/docs/math/matrix_exp.md b/docs/math/matrix_exp.md index f229a49..302947c 100644 --- a/docs/math/matrix_exp.md +++ b/docs/math/matrix_exp.md @@ -89,7 +89,9 @@ PyTorch factors this pattern through the helper 2. A. H. Al-Mohy and N. J. Higham, "Computing the Frechet Derivative of the Matrix Exponential," 2009. -## DB Status +## DB Families -`matrix_exp` is documented here as a known rule, but it is **not yet materialized** -in the current published `cases/` tree. + +### `matrix_exp/identity` + +The DB publishes the matrix exponential output directly. diff --git a/docs/math/registry.json b/docs/math/registry.json index eaede28..02efb67 100644 --- a/docs/math/registry.json +++ b/docs/math/registry.json @@ -457,6 +457,12 @@ "note_path": "docs/math/norm.md", "anchor": "family-matrix-norm-identity" }, + { + "op": "matrix_exp", + "family": "identity", + "note_path": "docs/math/matrix_exp.md", + "anchor": "family-identity" + }, { "op": "matrix_power", "family": "identity", diff --git a/generators/pytorch_v1.py b/generators/pytorch_v1.py index ce4ab35..e15fc2f 100644 --- a/generators/pytorch_v1.py +++ b/generators/pytorch_v1.py @@ -348,8 +348,32 @@ def _build_scalar_case_specs() -> tuple[CaseFamilySpec, ...]: return tuple(specs) +def _build_cmi_linalg_case_specs() -> tuple[CaseFamilySpec, ...]: + """Case specs for linalg ops whose OpInfo lives in common_methods_invocations.""" + return ( + CaseFamilySpec( + op="matrix_exp", + family="identity", + observable_kind="identity", + expected_behavior="success", + source_file="torch/testing/_internal/common_methods_invocations.py", + source_function="sample_inputs_matrix_exp", + upstream_name="matrix_exp", + upstream_variant_name="", + hvp_enabled=True, + inventory_kind="cmi_linalg", + supported_dtype_names=("float64", "complex128", "float32", "complex64"), + ), + ) + + def _build_case_specs() -> tuple[CaseFamilySpec, ...]: - return _build_success_case_specs() + _build_error_case_specs() + _build_scalar_case_specs() + return ( + _build_success_case_specs() + + _build_error_case_specs() + + _build_cmi_linalg_case_specs() + + _build_scalar_case_specs() + ) @lru_cache(maxsize=1) def _case_specs_cached() -> tuple[CaseFamilySpec, ...]: @@ -578,7 +602,7 @@ def _is_skippable_hvp_runtime_error(exc: RuntimeError) -> bool: def _resolve_runtime_for_spec(spec: CaseFamilySpec): - if spec.inventory_kind == "scalar": + if spec.inventory_kind in ("scalar", "cmi_linalg"): return import_scalar_generation_runtime() return import_generation_runtime() diff --git a/tests/test_math_registry.py b/tests/test_math_registry.py index a5bea6d..bce603a 100644 --- a/tests/test_math_registry.py +++ b/tests/test_math_registry.py @@ -356,12 +356,12 @@ def test_repo_scalar_ops_note_exposes_representative_op_anchors(self) -> None: self.assertEqual({"op-abs", "op-add", "op-sum", "op-var"} - anchors, set()) - def test_repo_matrix_exp_note_marks_db_status(self) -> None: + def test_repo_matrix_exp_note_has_db_family(self) -> None: text = ( Path(__file__).resolve().parents[1] / "docs" / "math" / "matrix_exp.md" ).read_text(encoding="utf-8") - self.assertIn("not yet materialized", text) + self.assertIn("family-identity", text) def test_repo_registry_contains_representative_family_mappings(self) -> None: root = Path(__file__).resolve().parents[1] diff --git a/validators/replay.py b/validators/replay.py index b39b09e..f5302fa 100644 --- a/validators/replay.py +++ b/validators/replay.py @@ -202,7 +202,7 @@ def _find_candidate_samples( def _prepare_samples_for_spec_dtype(torch, spec, *, dtype_name: str) -> list[PreparedSample]: - if getattr(spec, "inventory_kind", "linalg") == "scalar": + if getattr(spec, "inventory_kind", "linalg") in ("scalar", "cmi_linalg"): _, runtime_source = import_scalar_generation_runtime() else: _, runtime_source = import_generation_runtime() @@ -249,7 +249,7 @@ def _replay_success_case_for_sample( input_names = tuple(inputs.keys()) first_order = _first_order_comparison(comparison) second_order = _second_order_comparison(comparison) - if getattr(spec, "inventory_kind", "linalg") == "scalar": + if getattr(spec, "inventory_kind", "linalg") in ("scalar", "cmi_linalg"): _, runtime_source = import_scalar_generation_runtime() else: _, runtime_source = import_generation_runtime() From a43d53effd06bbd7c33f2c20b160f838c6873926 Mon Sep 17 00:00:00 2001 From: Hiroshi Shinaoka Date: Tue, 31 Mar 2026 20:31:08 +0900 Subject: [PATCH 3/4] Fix CI: add matrix_exp to complex-support ledger The complex-support ledger (merged from main) requires every published (op, family) to have an explicit entry. Add matrix_exp/identity as reviewed + covered and regenerate reports. Co-Authored-By: Claude Opus 4.6 (1M context) --- docs/generated/complex-support.md | 6 ++++-- docs/math/complex-support.json | 13 +++++++++++++ 2 files changed, 17 insertions(+), 2 deletions(-) diff --git a/docs/generated/complex-support.md b/docs/generated/complex-support.md index 5dcf3dd..04c27b5 100644 --- a/docs/generated/complex-support.md +++ b/docs/generated/complex-support.md @@ -4,8 +4,8 @@ Generated from the checked-in complex-support ledger and the published `cases/` ## Summary -- Total tracked families: 176 -- Ready for downstream: 103 +- Total tracked families: 177 +- Ready for downstream: 104 - Unsupported: 73 - Pending note review: 0 - Pending DB coverage: 0 @@ -89,6 +89,7 @@ Generated from the checked-in complex-support ledger and the published `cases/` | lu_factor | identity | not_required | covered | complex128, complex64 | - | yes | | lu_factor_ex | identity | not_required | covered | complex128, complex64 | - | yes | | lu_solve | identity | not_required | covered | complex128, complex64 | - | yes | +| matrix_exp | identity | reviewed | covered | complex128, complex64 | - | yes | | matrix_norm | identity | not_required | covered | complex128, complex64 | - | yes | | matrix_power | identity | reviewed | covered | complex128, complex64 | - | yes | | max_binary | identity | not_required | unsupported | - | float-only in pinned PyTorch upstream AD coverage | no | @@ -247,6 +248,7 @@ Generated from the checked-in complex-support ledger and the published `cases/` | lu_factor | identity | not_required | covered | complex128, complex64 | - | yes | | lu_factor_ex | identity | not_required | covered | complex128, complex64 | - | yes | | lu_solve | identity | not_required | covered | complex128, complex64 | - | yes | +| matrix_exp | identity | reviewed | covered | complex128, complex64 | - | yes | | matrix_norm | identity | not_required | covered | complex128, complex64 | - | yes | | matrix_power | identity | reviewed | covered | complex128, complex64 | - | yes | | mean | identity | reviewed | covered | complex128, complex64 | - | yes | diff --git a/docs/math/complex-support.json b/docs/math/complex-support.json index a7b3af4..0295f58 100644 --- a/docs/math/complex-support.json +++ b/docs/math/complex-support.json @@ -989,6 +989,19 @@ }, "unsupported_reason": null }, + { + "op": "matrix_exp", + "family": "identity", + "note": { + "path": "docs/math/matrix_exp.md", + "anchor": "family-identity", + "status": "reviewed" + }, + "db": { + "status": "covered" + }, + "unsupported_reason": null + }, { "op": "matrix_power", "family": "identity", From 087dd24d61e0cf7abbad30f6a7ae409a5085e0ab Mon Sep 17 00:00:00 2001 From: Hiroshi Shinaoka Date: Tue, 31 Mar 2026 20:41:53 +0900 Subject: [PATCH 4/4] Fix CI: route cmi_linalg through scalar tolerance resolver The upstream AD tolerance checker uses the same inventory_kind routing as the generator and replay validator. Extend both resolver dispatches to handle cmi_linalg alongside scalar. Co-Authored-By: Claude Opus 4.6 (1M context) --- scripts/check_upstream_ad_tolerances.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/scripts/check_upstream_ad_tolerances.py b/scripts/check_upstream_ad_tolerances.py index 63b5046..9a6d31b 100644 --- a/scripts/check_upstream_ad_tolerances.py +++ b/scripts/check_upstream_ad_tolerances.py @@ -72,7 +72,7 @@ def audit_against_upstream_ad_tolerances( spec = spec_index[(op, family)] resolver = ( resolve_upstream_scalar_ad_tolerance - if spec.inventory_kind == "scalar" + if spec.inventory_kind in ("scalar", "cmi_linalg") else resolve_upstream_ad_tolerance ) upstream = resolver( @@ -116,7 +116,7 @@ def audit_against_upstream_ad_tolerances( spec = spec_index[(op, family)] resolver = ( resolve_upstream_scalar_ad_tolerance - if spec.inventory_kind == "scalar" + if spec.inventory_kind in ("scalar", "cmi_linalg") else resolve_upstream_ad_tolerance ) upstream = resolver(