Fix explicit cast required

This is required for PyTorch which runs into an issue with narrowing types. Requires an explicit cast.
This commit is contained in:
Aaron Enye Shi
2018-07-20 16:34:15 -04:00
committato da GitHub
parent 5184a69410
commit 032ebd41ea
@@ -123,7 +123,7 @@ THE SOFTWARE.
typename std::enable_if<
(rank > 1) && sizeof...(Us) == rank>::type* = nullptr>
__host__ __device__
HIP_vector_type(Us... xs) noexcept { data = Native_vec_{xs...}; }
HIP_vector_type(Us... xs) noexcept { data = Native_vec_{static_cast<T>(xs)...}; }
__host__ __device__
HIP_vector_type(const HIP_vector_type&) = default;
__host__ __device__