Fix cxx11_tensor_{block_access, reduction} tests

This commit is contained in:
Eugene Zhulenev 2018-10-25 11:31:29 -07:00
parent 1dcf5a6ed8
commit 8a977c1f46
2 changed files with 5 additions and 3 deletions

View File

@ -820,8 +820,8 @@ static void test_block_cwise_binary_io_zero_strides() {
left_strides[4] = 0;
DSizes<Index, 5> right_sizes = rnd_dims;
right_sizes[1] = 0;
right_sizes[3] = 0;
right_sizes[1] = 1;
right_sizes[3] = 1;
DSizes<Index, 5> right_strides(ComputeStrides<Layout, 5>(right_sizes));
right_strides[1] = 0;

View File

@ -57,6 +57,8 @@ template <typename Scalar,int DataLayout>
static void test_simple_reductions() {
Tensor<Scalar, 4, DataLayout> tensor(2, 3, 5, 7);
tensor.setRandom();
// Add a little offset so that the product reductions won't be close to zero.
tensor += tensor.constant(Scalar(0.5f));
array<ptrdiff_t, 2> reduction_axis2;
reduction_axis2[0] = 1;
reduction_axis2[1] = 3;