torch-mlir

Commit Graph

Author	SHA1	Message	Date
Felix Schneider	0314188dbe	[torch] Basic support for per-channel quantized graphs (#3623 ) This patch adds basic support for lowering graphs with per-channel quantization. Per-channel quantized ops have to be excluded from `FuseQuantizedOps` for now but can be used in QDQ quantized form. Using this patch, we're able to import and execute (on the linalg backend) graphs with per-channel quantization applied using the "new" PyTorch 2.0 Export Quantization.	2024-08-10 15:51:09 +02:00
Aart Bik	fe836ceebf	[torch-mlir][test] cleanup trailing whitespace in mlir files (#2806 )	2024-01-25 14:24:13 -08:00
Rob Suderman	f6f890520b	[torch][quant] Quantized `torch.mm` for linalg with end-to-end test (#2750 ) This includes custom op matching for decomposed operations and fusing dequantization into dense operations. As a validation we compare to the dequant+mm torch implementation.	2024-01-24 14:02:50 -08:00

Author

SHA1

Message

Date

Felix Schneider

0314188dbe

[torch] Basic support for per-channel quantized graphs (#3623 )

This patch adds basic support for lowering graphs with per-channel
quantization. Per-channel quantized ops have to be excluded from
`FuseQuantizedOps` for now but can be used in QDQ quantized form.

Using this patch, we're able to import and execute (on the linalg
backend) graphs with per-channel quantization applied using the "new"
PyTorch 2.0 Export Quantization.

2024-08-10 15:51:09 +02:00

Aart Bik

fe836ceebf

[torch-mlir][test] cleanup trailing whitespace in mlir files (#2806 )

2024-01-25 14:24:13 -08:00

Rob Suderman

f6f890520b

[torch][quant] Quantized `torch.mm` for linalg with end-to-end test (#2750 )

This includes custom op matching for decomposed operations and fusing
dequantization into dense operations. As a validation we compare
to the dequant+mm torch implementation.

2024-01-24 14:02:50 -08:00

3 Commits (3f07077ff988e4ef870ce3d840391679e9acc03d)