ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-05-17 19:40:04 +00:00
Code Issues Packages Projects Releases Wiki Activity

adding ConstantMergedTensorDescriptor, refactering ConstantTensorDescriptor, Sequence

Browse Source 
[ROCm/composable_kernel commit: acd7082fe1]
This commit is contained in:
Chao Liu
2019-05-21 16:17:58 -05:00
parent 2c97c114cf
commit 45e1ad4dea
38 changed files with 1238 additions and 768 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
src/include/tensor.hpp
View File

@@ -93,7 +93,7 @@ struct TensorDescriptor
const std::vector<std::size_t>& GetStrides() const;
template <class... Is>
std::size_t Get1dIndex(Is... is) const
std::size_t GetOffsetFromMultiIndex(Is... is) const
{
assert(sizeof...(Is) == this->GetNumOfDimension());
std::initializer_list<std::size_t> iss{static_cast<std::size_t>(is)...};
@@ -246,13 +246,13 @@ struct Tensor
template <class... Is>
T& operator()(Is... is)
{
return mData[mDesc.Get1dIndex(is...)];
return mData[mDesc.GetOffsetFromMultiIndex(is...)];
}
template <class... Is>
const T& operator()(Is... is) const
{
return mData[mDesc.Get1dIndex(is...)];
return mData[mDesc.GetOffsetFromMultiIndex(is...)];
}
typename std::vector<T>::iterator begin() { return mData.begin(); }