Commit 0e32fd8
authored
Fix node output issues (microsoft#2497)
This pull request updates the `_quantize_matmul` function to improve
compatibility with downstream consumers by preserving the names of graph
outputs during quantization. Now, only intermediate output names are
modified, ensuring that external tools relying on specific output names
continue to work as expected.
Output name handling:
* Updated `_quantize_matmul` in
`olive/passes/onnx/kquant_quantization.py` to only rename intermediate
outputs, preserving the names of graph outputs so downstream consumers
(such as `genai_config.json`) remain compatible.1 parent 50e535f commit 0e32fd8
1 file changed
Lines changed: 4 additions & 1 deletion
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
380 | 380 | | |
381 | 381 | | |
382 | 382 | | |
383 | | - | |
| 383 | + | |
| 384 | + | |
| 385 | + | |
| 386 | + | |
384 | 387 | | |
385 | 388 | | |
386 | 389 | | |
| |||
0 commit comments