Skip to content

Fix JAX dot out handling#3387

Closed
FlorianPfaff wants to merge 2 commits into
mainfrom
fix-jax-dot-out-contract-20260629
Closed

Fix JAX dot out handling#3387
FlorianPfaff wants to merge 2 commits into
mainfrom
fix-jax-dot-out-contract-20260629

Conversation

@FlorianPfaff

Copy link
Copy Markdown
Owner

Summary

  • patch the JAX backend facade so backend.dot(..., out=...) accepts NumPy-style output arrays
  • keep the raw pyrecest._backend.jax.dot entry point aligned after PyRecEst import
  • validate out shape before returning the JAX immutable update result
  • add subprocess regression coverage for scalar, vector, raw-backend, and incompatible-shape cases

Bug fixed

Current main defines the JAX backend dot helper as dot(a, b) only. Code using the NumPy-compatible out keyword therefore fails with TypeError under PYRECEST_BACKEND=jax, even though other PyRecEst backend helpers expose NumPy-style output handling.

Testing

  • Syntax-checked the changed source and test content locally before committing.
  • Added tests/backend_support/test_jax_dot_out_contract.py; full CI should run on GitHub.

@FlorianPfaff FlorianPfaff enabled auto-merge (squash) June 29, 2026 09:56
@github-actions

Copy link
Copy Markdown
Contributor

MegaLinter analysis: Success

Descriptor Linter Files Fixed Errors Warnings Elapsed time
✅ COPYPASTE jscpd yes no no 70.62s
✅ JSON prettier 7 0 0 0 0.95s
✅ JSON v8r 7 0 0 4.75s
✅ MARKDOWN markdownlint 68 0 0 0 2.1s
✅ MARKDOWN markdown-table-formatter 68 0 0 0 0.76s
✅ PYTHON black 1095 0 0 0 61.49s
✅ PYTHON isort 1095 0 0 0 5.67s
✅ REPOSITORY checkov yes no no 47.39s
✅ REPOSITORY gitleaks yes no no 9.34s
✅ REPOSITORY git_diff yes no no 0.21s
✅ REPOSITORY secretlint yes no no 33.13s
✅ REPOSITORY syft yes no no 4.77s
✅ REPOSITORY trivy-sbom yes no no 12.27s
✅ REPOSITORY trufflehog yes no no 21.0s
✅ YAML prettier 11 0 0 0 0.81s
✅ YAML v8r 11 0 0 11.4s
✅ YAML yamllint 11 0 0 0.57s

Notices

📣 MegaLinter 9.5.0 is out! Discover the new features and security recommendations in the release announcement. (Skip this info by defining SECURITY_SUGGESTIONS: false)

See detailed reports in MegaLinter artifacts

Your project could benefit from a custom flavor, which would allow you to run only the linters you need, and thus improve runtime performances. (Skip this info by defining FLAVOR_SUGGESTIONS: false)

  • Documentation: Custom Flavors
  • Command: npx mega-linter-runner@9.5.0 --custom-flavor-setup --custom-flavor-linters PYTHON_BLACK,PYTHON_ISORT,COPYPASTE_JSCPD,JSON_V8R,JSON_PRETTIER,MARKDOWN_MARKDOWNLINT,MARKDOWN_MARKDOWN_TABLE_FORMATTER,REPOSITORY_CHECKOV,REPOSITORY_GIT_DIFF,REPOSITORY_GITLEAKS,REPOSITORY_SECRETLINT,REPOSITORY_SYFT,REPOSITORY_TRIVY_SBOM,REPOSITORY_TRUFFLEHOG,YAML_PRETTIER,YAML_YAMLLINT,YAML_V8R

MegaLinter is graciously provided by OX Security
Show us your support by starring ⭐ the repository

auto-merge was automatically disabled June 29, 2026 11:42

Pull request was closed

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant