Changed implicit cast to explicit cast

apache · Aug 31, 2018 · 93c105a · 93c105a
1 parent 61a2cd7
commit 93c105a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/python/unittest/test_operator.py b/tests/python/unittest/test_operator.py
@@ -1650,7 +1650,7 @@ def test_depthwise_convolution():
                             exe2 = y2.simple_bind(mx.cpu(), x=shape, w=(num_filter, shape[1]//num_group)+kernel,
                                     b=(num_filter,))
                             for arr1, arr2 in zip(exe1.arg_arrays, exe2.arg_arrays):
-                                arr1[:] = np.random.normal(size=arr1.shape)
+                                arr1[:] = np.float32(np.random.normal(size=arr1.shape))
                                 arr2[:] = arr1
                             exe1.forward(is_train=True)
                             exe1.backward(exe1.outputs[0])