SIGN IN SIGN UP

DQLinear: Add support for per_token activation and fp16 dtype (#2327)

Summary:
Pull Request resolved: https://github.com/pytorch/executorch/pull/2327

Wire per_token quantization support in xnnpack delegate. ATM PT2e quant flow (XNNPACKQuantizer --> prepare --> convert) doesn't support `q/dq` nodes for `per_token`. So added eager mode hacks to generate such graph in tests. Once we add support the graph should remain the same so it is ok to test lowering logic like that.

bypass-github-export-checks

Reviewed By: kimishpatel, mcr229

Differential Revision: D54026295

fbshipit-source-id: 269559d321674b21f18352c7e1ed649665963567
D
Digant Desai committed
45df800520f756cc8abded8d5c4c4d9782907fc6
Parent: 588c391
Committed by Facebook GitHub Bot <facebook-github-bot@users.noreply.github.com> on 3/18/2024, 3:56:53 AM