Fix bug in a test + compilation errors

This commit is contained in:
Eugene Zhulenev
2018-08-09 09:44:07 -07:00
parent 1c8b9e10a7
commit cfaedb38cd
4 changed files with 13 additions and 11 deletions

View File

@@ -317,7 +317,7 @@ static void test_execute_reshape(Device d)
DSizes<Index, ReshapedDims> reshaped_dims;
reshaped_dims[shuffle[0]] = dims[0] * dims[1];
for (int i = 2; i < NumDims; ++i) reshaped_dims[shuffle[i]] = dims[i];
for (int i = 1; i < ReshapedDims; ++i) reshaped_dims[shuffle[i]] = dims[i + 1];
Tensor<T, ReshapedDims, Options, Index> golden = src.reshape(reshaped_dims);