IntenseWebs/openvino
3
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fix sporadic fails when beta==0 in baddmm (#16610)

Browse Source 
* Fix sporadic fails when beta==0 in baddmm

* Remove sporadic test loop
This commit is contained in:
Maxim Vafin 2023-03-28 20:47:35 +02:00 committed by GitHub
parent 4403433309
commit 8d90c11a35
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 8 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
tests/layer_tests/pytorch_tests/test_addmm.py
View File

@ -83,8 +83,14 @@ class TestBAddBMM(PytorchLayerTest):
])
@pytest.mark.parametrize("alpha,beta",
[(1., 1.), (0., 1.), (1., 0.), (1., 2.), (2., 1.), (-5., -6.), (3., 4.), (0.5, 0.75),
(1, 1)])
[ # beta==0 in some cases produce nan in pytorch
(1., 1.),
(0., 1.),
(-5., -6.),
(3., 4.),
(0.5, 0.75),
(1, 1)
])
@pytest.mark.nightly
@pytest.mark.precommit
def test_baddbmm(self, kwargs_to_prepare_input, alpha, beta, ie_device, precision, ir_version):