Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Introduce internal interface for ValuePropBackend #201

Open
wants to merge 7 commits into
base: main
Choose a base branch
from

Conversation

neNasko1
Copy link
Contributor

@neNasko1 neNasko1 commented Jan 15, 2025

Since we currently "bless" the reference and onnxruntime implementations as the only way to propagate value internally, this PR refactors a bit the internals to allow a better interface for "custom" execution providers(i.e. @cbourjau rust runtime, onnx-mlir...).
Consult the tests for a POC which could also be used to register custom opsets as well.

An additional internal change is related to the PropDict type as previously it was too wide and didn't capture the correct semantics of the returned values from onnxruntime. As per the definition of from_ref_value we can see that the returned feed is of type dict[str, PropValue] which makes more sense to model around.

This exposed a bug in the function value inference which was unnoticed and is now fixed.

Checklist

  • Added a CHANGELOG.rst entry

@neNasko1 neNasko1 marked this pull request as ready for review January 15, 2025 14:02
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