Skip to content

Add extension AD rule registry - #846

Merged
shinaoka merged 1 commit into
mainfrom
codex/extension-rule-registry
May 5, 2026
Merged

Add extension AD rule registry#846
shinaoka merged 1 commit into
mainfrom
codex/extension-rule-registry

Conversation

@shinaoka

@shinaoka shinaoka commented May 5, 2026

Copy link
Copy Markdown
Member

Summary

  • Add a separate ExtensionAdRule registry for out-of-tree tensor operations, with fallible linearize/transpose dispatch through chainrules/tidu.
  • Re-export registration helpers and ExtensionFamilyId, and add a proc-macro crate for stable family IDs.
  • Add traced and eager error propagation for missing extension AD rules, plus facade tests for registered-rule AD and validation paths.
  • Update the extension-op spec to describe registered AD rules and same-graph extension composition.

Dependencies

Validation

  • cargo fmt --all --check
  • cargo nextest run --workspace --release --no-fail-fast
  • cargo test --doc --workspace --release
  • cargo llvm-cov nextest --workspace --release --json --output-path coverage.json
  • python3 scripts/check-coverage.py coverage.json
  • cargo doc --workspace --no-deps
  • python3 scripts/check-docs-site.py
  • bash scripts/check-repo-settings.sh

Implemented with OpenAI Codex.

@shinaoka
shinaoka enabled auto-merge (squash) May 5, 2026 08:40
@shinaoka
shinaoka merged commit f33337e into main May 5, 2026
8 checks passed
@shinaoka
shinaoka deleted the codex/extension-rule-registry branch May 5, 2026 08:46
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