mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-07 12:21:27 +01:00
fix hypothesis timeout (#26280)
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/26280 ghstack-source-id: 90160270 Test Plan: testinprod Differential Revision: D17396861 fbshipit-source-id: ee2348ffa7f6092e2c5647a42d0e17879dcfacd0
This commit is contained in:
parent
38b2bc1451
commit
c006356034
|
|
@ -137,6 +137,7 @@ class TestQuantizedOps(TestCase):
|
|||
self.assertEqual(qY, qY_hat, message="{} relu failed".format(name))
|
||||
|
||||
"""Tests the correctness of the scalar addition."""
|
||||
@no_deadline
|
||||
@given(A=hu.tensor(shapes=hu.array_shapes(1, 4, 1, 5),
|
||||
elements=st.floats(-1e6, 1e6, allow_nan=False),
|
||||
qparams=hu.qparams()),
|
||||
|
|
|
|||
Loading…
Reference in New Issue
Block a user