Adding constants from [PR418](NVIDIA/cub#418).

ROCm · Mar 4, 2022 · 971038e · 971038e
1 parent e0040e2
commit 971038e
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/hipcub/include/hipcub/backend/rocprim/util_type.hpp b/hipcub/include/hipcub/backend/rocprim/util_type.hpp
@@ -364,10 +364,9 @@ struct Uninitialized
     /// Biggest memory-access word that T is a whole multiple of and is not larger than the alignment of T
     typedef typename UnitWord<T>::DeviceWord DeviceWord;
 
-    enum
-    {
-        WORDS = sizeof(T) / sizeof(DeviceWord)
-    };
+    static constexpr std::size_t DATA_SIZE = sizeof(T);
+    static constexpr std::size_t WORD_SIZE = sizeof(DeviceWord);
+    static constexpr std::size_t WORDS = DATA_SIZE / WORD_SIZE;
 
     /// Backing storage
     DeviceWord storage[WORDS];