Add seed input to GeneratorTensor_4 for normal_distribution generator

22214cae · Qianfeng Zhang · 0aa85235 · 22214cae
Commit 22214cae authored Oct 17, 2023 by Qianfeng Zhang
Hide whitespace changes
Inline Side-by-side

Showing with 3 additions and 2 deletions

library/include/ck/library/utility/host_tensor_generator.hpp library/include/ck/library/utility/host_tensor_generator.hpp +3 -2

No files found.
--- a/library/include/ck/library/utility/host_tensor_generator.hpp
+++ b/library/include/ck/library/utility/host_tensor_generator.hpp
@@ -130,10 +130,11 @@ struct GeneratorTensor_3<ck::bhalf_t>
 template <typename T>
 struct GeneratorTensor_4
 {
-    std::default_random_engine generator;
+    std::mt19937 generator;
    std::normal_distribution<float> distribution;
-    GeneratorTensor_4(float mean, float stddev) : generator(1), distribution(mean, stddev){};
+    GeneratorTensor_4(float mean, float stddev, unsigned int seed = 1)
+        : generator(seed), distribution(mean, stddev){};
    template <typename... Is>
    T operator()(Is...)