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

[Thresholding] Generalize data layouts for node execution #1289

Open
wants to merge 1 commit into
base: dev
Choose a base branch
from

Conversation

iksnagreb
Copy link
Contributor

See fastmachinelearning/qonnx#143 for the similar generalization applied to QONNX MultiThreshold

See fastmachinelearning/qonnx#143 for the
similar generalization applied to QONNX MultiThreshold
@auphelia
Copy link
Collaborator

auphelia commented Mar 4, 2025

Does this PR depend on fastmachinelearning/qonnx#143 ? Or is it related but independent?

@iksnagreb
Copy link
Contributor Author

No, it does not depend on the QONNX PR, this mirrors the changes to the QONNX MultiThreshold node execution for the FINN custom-op Thresholding node execution. This is also still more of a workaround, we do not even have access to the layout annotations here so it is even less generalized and always guesses... Merging this should not break anything, behavior of the existing 2d and 4d layouts is the same, but as also proposed by @maltanar at the QONNX PR, we should seriously consider more robust alternatives to the data layout strings...

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.

2 participants