frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 12:54:11 +08:00
Code Issues Packages Projects Releases Wiki Activity

Revert "[Quant][CPU] fix fake_quantize_per_tensor_affine of inf values (#155109)"

Browse Source 
This reverts commit e375d21bb9b0ef6fefe7a8af5a054a17de8c63c9.

Reverted https://github.com/pytorch/pytorch/pull/155109 on behalf of https://github.com/malfet due to Looks like it broke ROCM tests ([comment](https://github.com/pytorch/pytorch/pull/155109#issuecomment-2977428354))
This commit is contained in:
PyTorch MergeBot
2025-06-16 17:22:53 +00:00
parent 45596ec58f
commit e9fdaf8701
2 changed files with 6 additions and 23 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
test/quantization/core/test_workflow_ops.py
View File

@ -1038,21 +1038,6 @@ class TestFakeQuantizeOps(TestCase):
input, scale, zero_point, axis, quant_min, quant_max
)
@skipIfTorchDynamo("Not a suitable test for TorchDynamo")
@given(dtype=st.sampled_from([torch.float, torch.float64, torch.half, torch.bfloat16]),
device=st.sampled_from(['cpu', 'cuda'] if torch.cuda.is_available() else ['cpu']))
def test_fake_quantize_per_tensor_affine_inf(self, dtype, device) -> None:
# https://github.com/pytorch/pytorch/issues/154328
input_tensor = torch.tensor([torch.inf], dtype=dtype).to(device)
scale = 0.01
zero_point = 0
quant_min = 0
quant_max = 255
result = torch.fake_quantize_per_tensor_affine(input_tensor, scale, zero_point, quant_min, quant_max)
ref_result = (min(quant_max, max(quant_min, torch.round(input_tensor / scale) + zero_point)) - zero_point) * scale
ref_result = torch.Tensor([ref_result]).to(dtype).to(device)
self.assertEqual(result, ref_result)
class TestFusedObsFakeQuant(TestCase):
@given(device=st.sampled_from(['cpu', 'cuda'] if torch.cuda.is_available() else ['cpu']),