Skip to content

Brevitas: make QDQ ONNX export optional#94

Open
fxmarty wants to merge 2 commits intomainfrom
optional-qdq
Open

Brevitas: make QDQ ONNX export optional#94
fxmarty wants to merge 2 commits intomainfrom
optional-qdq

Conversation

@fxmarty
Copy link
Copy Markdown
Contributor

@fxmarty fxmarty commented Mar 1, 2024

As per title.

@Giuseppe5
Copy link
Copy Markdown
Contributor

Shall we hide the export phase in a function?
Considering that soon we will also introduce a flag to do post-processing for transforming QCDQ to QOp (at least for the short term).

@fxmarty
Copy link
Copy Markdown
Contributor Author

fxmarty commented Mar 1, 2024

@Giuseppe5 yes it would be good indeed.

@Giuseppe5
Copy link
Copy Markdown
Contributor

Giuseppe5 commented Mar 2, 2024

The possibility to exclude Q nodes from export it's supported only for Pytorch 2.2+, and I suspect that it is because of this PR:
pytorch/pytorch#111097

For now, I would add a constraints to use PyTorch > 2.2+.

In parallel, it could be worth exploring if we can control what is a initializer and what is a parameter when exporting a graph from pytorch to onnx.

@Giuseppe5 Giuseppe5 mentioned this pull request Mar 14, 2024
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