OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-06 12:20:52 +01:00
Code Issues Actions 38 Packages Projects Releases Wiki Activity

Enable UBSAN test (#147511)

Browse Source 
Fixes #ISSUE_NUMBER

Pull Request resolved: https://github.com/pytorch/pytorch/pull/147511
Approved by: https://github.com/colesbury
This commit is contained in:
cyy 2025-03-07 00:35:30 +00:00 committed by PyTorch MergeBot
parent 33a285379a
commit 50eb4f3990
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
test/quantization/core/test_quantized_op.py
View File

@ -25,7 +25,7 @@ hu.assert_deadline_disabled()
from torch.testing._internal.common_cuda import SM80OrLater
from torch.testing._internal.common_utils import TestCase
from torch.testing._internal.common_utils import IS_PPC, TEST_WITH_UBSAN, IS_MACOS, IS_SANDCASTLE, IS_FBCODE, IS_ARM64
from torch.testing._internal.common_utils import IS_PPC, IS_MACOS, IS_SANDCASTLE, IS_FBCODE, IS_ARM64
from torch.testing._internal.common_quantization import skipIfNoFBGEMM, skipIfNoQNNPACK, skipIfNoONEDNN
from torch.testing._internal.common_quantized import _quantize, _dequantize, _calculate_dynamic_qparams, \
override_quantized_engine, supported_qengines, override_qengines, _snr
@ -3698,7 +3698,7 @@ class TestDynamicQuantizedOps(TestCase):
# The goal here is to show that the dynamic op is the same as
# calc params->quantize_input->quantized op->dequantize output
if qengine_is_qnnpack() and (IS_PPC or TEST_WITH_UBSAN):
if qengine_is_qnnpack() and IS_PPC:
return # not supported by QNNPACK
if qengine_is_qnnpack():
@ -5720,7 +5720,7 @@ class TestQuantizedConv(TestCase):
def test_qconv_transpose1d(self):
if not qengine_is_qnnpack():
return # Currently only the QNNPACK is supported
if qengine_is_qnnpack() and (IS_PPC or TEST_WITH_UBSAN):
if qengine_is_qnnpack() and IS_PPC:
return # QNNPACK doesn't support these
batch_size = 2
input_channels_per_group_list = [2, 32]
@ -5865,7 +5865,7 @@ class TestQuantizedConv(TestCase):
Y_scale,
Y_zero_point,
use_bias):
if qengine_is_qnnpack() and (IS_PPC or TEST_WITH_UBSAN):
if qengine_is_qnnpack() and IS_PPC:
return # QNNPACK doesn't support these
# ONEDNN does not support output paddings
if qengine_is_onednn() and (o_pad_h, o_pad_w) != (0, 0):