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