Put random number gen to a separate header

include/ck/utility/data_type.hpp

@@ -3,7 +3,6 @@
 
 #pragma once
 
-#include "ck/utility/f8_utils.hpp"
 #include "ck/utility/statically_indexed_array.hpp"
 #include "ck/utility/type_convert.hpp"
 

include/ck/utility/f8_utils.hpp

@@ -3,8 +3,6 @@
 
 #pragma once
 
-#include "ck/utility/statically_indexed_array.hpp"
-
 namespace ck {
 
 using f8_t   = uint8_t;

include/ck/utility/random_gen.hpp

+// SPDX-License-Identifier: MIT
+// Copyright (c) 2018-2023, Advanced Micro Devices, Inc. All rights reserved.
+
+#pragma once
+
+namespace ck {
+
+// Pseudo random number generator
+// version for fp32
+template <typename T, uint32_t seed_t, std::enable_if_t<std::is_same<float, T>{}, bool> = false>
+__host__ __device__ uint32_t prand_generator(index_t id, T val, uint32_t seed = seed_t)
+{
+    uint32_t x         = *(reinterpret_cast<uint32_t*>(&val));
+    uint32_t drop_bits = uint32_t(x) & 0xFFFFu;
+    drop_bits ^= x >> 16;
+    drop_bits = ((drop_bits & 31) << 11) | (drop_bits >> 5);
+    drop_bits *= 0x7000149;
+    // NOTE: If id is in 64 bit, we are only using lower 32 bit.
+    //       So, it can have an effect of using same id for multiple elements when the id is very
+    //       large!
+    uint32_t rng = (drop_bits ^ 0x13371337 ^ (id * 229791) ^ seed);
+    return rng;
+}
+
+// version for fp16
+template <typename T, uint32_t seed_t, std::enable_if_t<std::is_same<half_t, T>{}, bool> = false>
+__host__ __device__ uint32_t prand_generator(index_t id, T val, uint32_t seed = seed_t)
+{
+    uint16_t x         = *(reinterpret_cast<uint16_t*>(&val));
+    uint32_t drop_bits = uint32_t(x) & 0xFFFFu;
+    drop_bits          = ((drop_bits & 31) << 11) | (drop_bits >> 5);
+    drop_bits *= 0x7000149;
+    // NOTE: If id is in 64 bit, we are only using lower 32 bit.
+    //       So, it can have an effect of using same id for multiple elements when the id is very
+    //       large!
+    uint32_t rng = (drop_bits ^ 0x13371337 ^ (id * 229791) ^ seed);
+    return rng;
+}
+
+// return 0 if data is not fp16 or fp32
+template <typename T,
+          uint32_t seed_t,
+          std::enable_if_t<!(std::is_same<float, T>{} || std::is_same<half_t, T>{}), bool> = false>
+__host__ __device__ uint32_t prand_generator(int id, T val, uint32_t seed = seed_t)
+{
+    std::ignore = id;
+    std::ignore = val;
+    std::ignore = seed;
+
+    return 0;
+}
+
+} // namespace ck

include/ck/utility/type_convert.hpp

@@ -4,6 +4,7 @@
 #pragma once
 
 #include "ck/utility/f8_utils.hpp"
+#include "ck/utility/random_gen.hpp"
 
 namespace ck {
 
@@ -198,51 +199,6 @@ inline __host__ __device__ constexpr bhalf_t bf16_convert_rtn<bhalf_t, half_t>(h
     return bf16_convert_rtn<bhalf_t>(x_fp32);
 }
 
-// Pseudo random number generator
-// version for fp32
-template <typename T, uint32_t seed_t, std::enable_if_t<std::is_same<float, T>{}, bool> = false>
-__host__ __device__ uint32_t prand_generator(index_t id, T val, uint32_t seed = seed_t)
-{
-    uint32_t x         = *(reinterpret_cast<uint32_t*>(&val));
-    uint32_t drop_bits = uint32_t(x) & 0xFFFFu;
-    drop_bits ^= x >> 16;
-    drop_bits = ((drop_bits & 31) << 11) | (drop_bits >> 5);
-    drop_bits *= 0x7000149;
-    // NOTE: If id is in 64 bit, we are only using lower 32 bit.
-    //       So, it can have an effect of using same id for multiple elements when the id is very
-    //       large!
-    uint32_t rng = (drop_bits ^ 0x13371337 ^ (id * 229791) ^ seed);
-    return rng;
-}
-
-// version for fp16
-template <typename T, uint32_t seed_t, std::enable_if_t<std::is_same<half_t, T>{}, bool> = false>
-__host__ __device__ uint32_t prand_generator(index_t id, T val, uint32_t seed = seed_t)
-{
-    uint16_t x         = *(reinterpret_cast<uint16_t*>(&val));
-    uint32_t drop_bits = uint32_t(x) & 0xFFFFu;
-    drop_bits          = ((drop_bits & 31) << 11) | (drop_bits >> 5);
-    drop_bits *= 0x7000149;
-    // NOTE: If id is in 64 bit, we are only using lower 32 bit.
-    //       So, it can have an effect of using same id for multiple elements when the id is very
-    //       large!
-    uint32_t rng = (drop_bits ^ 0x13371337 ^ (id * 229791) ^ seed);
-    return rng;
-}
-
-// return 0 if data is not fp16 or fp32
-template <typename T,
-          uint32_t seed_t,
-          std::enable_if_t<!(std::is_same<float, T>{} || std::is_same<half_t, T>{}), bool> = false>
-__host__ __device__ uint32_t prand_generator(int id, T val, uint32_t seed = seed_t)
-{
-    std::ignore = id;
-    std::ignore = val;
-    std::ignore = seed;
-
-    return 0;
-}
-
 // Declare a template function for fp8 conversion using SR
 template <typename Y, typename X>
 __host__ __device__ constexpr Y f8_convert_sr(X x);