Force copying parameters in 'Embed' ctor

18495464 · Po-Yen, Chen · b076a02a · 18495464
Commit 18495464 authored May 07, 2023 by Po-Yen, Chen
Show whitespace changes
Inline Side-by-side

Showing with 3 additions and 3 deletions

include/ck/tensor_description/multi_index_transform.hpp include/ck/tensor_description/multi_index_transform.hpp +3 -3

No files found.
--- a/include/ck/tensor_description/multi_index_transform.hpp
+++ b/include/ck/tensor_description/multi_index_transform.hpp
@@ -393,9 +393,9 @@ struct Embed

    __host__ __device__ constexpr Embed() = default;

-    __host__ __device__ constexpr Embed(const UpLengths& up_lengths,
-                                        const Coefficients& coefficients)
-        : up_lengths_{up_lengths}, coefficients_{coefficients}
+    /// NOTE: force copying here to prevent uninitialized data members (on device side)
+    __host__ __device__ constexpr Embed(UpLengths up_lengths, Coefficients coefficients)
+        : up_lengths_{std::move(up_lengths)}, coefficients_{std::move(coefficients)}
    {
    }