This is an automated email from the ASF dual-hosted git repository.
bgawrych pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/incubator-mxnet.git
The following commit(s) were added to refs/heads/master by this push:
new 08f578b946 Fix test_bf16_binary_broadcast_elemwise_mixed_input (#20986)
08f578b946 is described below
commit 08f578b9462208be89f58ed25d653deb099b21cf
Author: AdamGrabowski <[email protected]>
AuthorDate: Tue Jun 21 08:53:09 2022 +0200
Fix test_bf16_binary_broadcast_elemwise_mixed_input (#20986)
---
tests/python/dnnl/test_bf16_operator.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/tests/python/dnnl/test_bf16_operator.py
b/tests/python/dnnl/test_bf16_operator.py
index 631e34f5ff..13ff824bfc 100644
--- a/tests/python/dnnl/test_bf16_operator.py
+++ b/tests/python/dnnl/test_bf16_operator.py
@@ -217,8 +217,8 @@ def test_bf16_binary_broadcast_elemwise_funcs():
@pytest.mark.parametrize('function', [mx.np.add, mx.np.subtract,
mx.np.multiply, mx.np.divide])
@pytest.mark.parametrize('dtype', [np.float32, np.float64])
[email protected]('ndim', [1,2,3,4,5,6])
-def test_bf16_binary_broadcast_elemwise_mixed_input(function, dtype, ndim):
+def test_bf16_binary_broadcast_elemwise_mixed_input(function, dtype):
+ ndim = np.random.randint(1, 6)
dshape_0 = rand_shape_nd(ndim)
dshape_1 = tuple()
for i in range(ndim):
