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

Merge commit '77123600ee4b6fae077a2145b68b00a8b2ce9460' into develop

Browse Source
This commit is contained in:
assistant-librarian[bot]
2025-06-24 15:21:28 +00:00
parent 845e97c29d
commit 357a7be3ba
19 changed files with 1717 additions and 62 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
include/ck_tile/host/host_tensor.hpp
View File

@@ -230,7 +230,7 @@ struct HostTensorDescriptor
* @param iss Vector containing the multi-dimensional indices
* @return The calculated linear offset as a size_t
*/
std::size_t GetOffsetFromMultiIndex(std::vector<std::size_t> iss) const
std::size_t GetOffsetFromMultiIndex(const std::vector<std::size_t>& iss) const
{
return std::inner_product(iss.begin(), iss.end(), mStrides.begin(), std::size_t{0});
}
@@ -540,9 +540,12 @@ struct HostTensor
return mData[GetOffsetFromMultiIndex(is...)];
}
T& operator()(std::vector<std::size_t> idx) { return mData[GetOffsetFromMultiIndex(idx)]; }
T& operator()(const std::vector<std::size_t>& idx)
{
return mData[GetOffsetFromMultiIndex(idx)];
}
const T& operator()(std::vector<std::size_t> idx) const
const T& operator()(const std::vector<std::size_t>& idx) const
{
return mData[GetOffsetFromMultiIndex(idx)];
}