ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-06-05 20:55:59 +00:00
Code Issues Packages Projects Releases Wiki Activity

Allow conversion between Tensor<> specializations

Browse Source
This commit is contained in:
Po-Yen, Chen
2022-08-19 00:30:53 -04:00
parent 30ed3e218c
commit cbbe2485b2
1 changed files with 3 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
library/include/ck/library/utility/host_tensor.hpp
View File

@@ -266,6 +266,9 @@ struct Tensor
Tensor(const Tensor& other) : mDesc(other.mDesc), mData(other.mData) {}
template <typename OtherT, typename = std::enable_if_t<sizeof(T) == sizeof(OtherT) && std::is_convertible_v<T, OtherT>>>
Tensor(const Tensor<OtherT>& other) : mDesc(other.mDesc), mData(std::begin(other.mData), std::end(other.mData)) {}
Tensor& operator=(const Tensor& other)
{
mDesc = other.mDesc;