Skip to content

feat(xla): bind operator numeric contracts - #937

Merged
inureyes merged 1 commit into
mainfrom
feature/issue-932-operator-contracts
Jul 27, 2026
Merged

inureyes merged 1 commit into
mainfrom
feature/issue-932-operator-contracts

Conversation

@inureyes

@inureyes inureyes commented Jul 27, 2026 •

Copy link
Copy Markdown
Member

Summary

  • define a stable per-operator numeric descriptor covering checkpoint encoding, input/result materialization, accumulator dtype, rounding boundaries, association policy, exact affine dequantization order, and backend identity
  • add auxiliary artifact schema v3 and bind the operator descriptor fingerprint to write, load, mismatch rejection, and downgrade protection
  • preserve the exact v1/v2 artifact digests and make an existing diagnostics-only Gemma3n SDPA ABI test runnable on CPU-IREE

Safety boundary

Validation

  • cargo check -p mlxcel-xla
  • cargo test -p mlxcel-xla --features iree operator_numeric_contract -- --nocapture (4 passed)
  • cargo test -p mlxcel-xla --features iree aux_manifest::tests -- --nocapture (14 passed)
  • cargo test -p mlxcel-xla --features diagnostics sdpa_production_abi_passes_dynamic_position_and_exact_scale_bits -- --nocapture (1 passed)
  • cargo-clippy clippy -p mlxcel-xla --features diagnostics --tests --no-deps (passed; existing warnings only)
  • cargo-fmt fmt --all -- --check

Refs #932

Add a versioned per-operator numeric descriptor and bind it to auxiliary artifact schema v3. Preserve v1/v2 digests, reject mismatches and downgrades, and keep production family activation unchanged.\n\nRefs #932
@inureyes inureyes added status:review Under review type:enhancement New features, capabilities, or significant additions priority:high High priority area:architecture Architecture and code structure changes area:core mlxcel-core: MLX FFI, primitives, KV cache, layers area:inference Generation, sampling, decoding (incl. speculative, DRY) labels Jul 27, 2026
@inureyes
inureyes merged commit c2bbff6 into main Jul 27, 2026
5 checks passed
@inureyes
inureyes deleted the feature/issue-932-operator-contracts branch July 27, 2026 00:42
@inureyes inureyes added status:done Completed and removed status:review Under review labels Jul 27, 2026
@inureyes inureyes self-assigned this Aug 31, 2026
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

area:architecture Architecture and code structure changes area:core mlxcel-core: MLX FFI, primitives, KV cache, layers area:inference Generation, sampling, decoding (incl. speculative, DRY) priority:high High priority status:done Completed type:enhancement New features, capabilities, or significant additions

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant