address review comments

zheyuye · Feb 16, 2020 · 56d873f · 56d873f
1 parent 38070d9
commit 56d873f
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/cpp/include/test_mkldnn.h b/tests/cpp/include/test_mkldnn.h
@@ -70,9 +70,9 @@ inline static void InitDefaultArray(NDArray *arr, bool is_rand = false, int max
 
  for (int i = 0; i < size; i++)
  if (is_rand) {
- data[i] = (std::rand() % (max * 2)) - max;
+ data[i] = std::rand() * 1.0f / RAND_MAX * max * 2 - max;
  } else {
- data[i] = i % (max * 2) - max;
+ data[i] = i * 1.0f / size * max * 2 - max;
  }
 }