kausik 4f60a2e39c Set correct output dtype for dequantize op during convert_pt2e in decomposed mode (#128953) ...

Earlier the signature of dequantize ops for decomposed quantized Tensor was changed for wider use-cases where the output dtype can be different from torch.float and needs to be passed during dequantization.
Please refer: https://github.com/pytorch/pytorch/pull/121450

However, setting of correct output dtype for dequantize ops was still missing in convert_pt2e flow.

This change enables the users to use PT2E quantization flow with non torch.float unquantized dtype, such as torch.bfloat16.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/128953
Approved by: https://github.com/jgong5, https://github.com/jerryzh168

2024-07-19 04:58:02 +00:00

..

nn

[Docs] Update links (#128795)

2024-06-17 14:55:32 +00:00

ns

Flip default value for mypy disallow_untyped_defs [3/11] (#127840)

2024-06-08 18:28:01 +00:00

pruning

Flip default value for mypy disallow_untyped_defs [4/11] (#127841)

2024-06-08 18:36:48 +00:00

quantization

Set correct output dtype for dequantize op during convert_pt2e in decomposed mode (#128953)

2024-07-19 04:58:02 +00:00

__init__.py

Flip default value for mypy disallow_untyped_defs [3/11] (#127840)

2024-06-08 18:28:01 +00:00