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

Merge commit 'b0a2d99d100f2e4212ebbed080acb49a404035ab' into develop

Browse Source
This commit is contained in:
assistant-librarian[bot]
2025-09-26 01:40:00 +00:00
parent e575ac4332
commit 19f49ee63e
123 changed files with 1733 additions and 849 deletions
Download Patch File Download Diff File Expand all files Collapse all files

23
library/src/utility/host_tensor.cpp
View File

@@ -5,18 +5,6 @@
#include "ck/library/utility/host_tensor.hpp"
void HostTensorDescriptor::CalculateStrides()
{
mStrides.clear();
mStrides.resize(mLens.size(), 0);
if(mStrides.empty())
return;
mStrides.back() = 1;
std::partial_sum(
mLens.rbegin(), mLens.rend() - 1, mStrides.rbegin() + 1, std::multiplies<std::size_t>());
}
std::size_t HostTensorDescriptor::GetNumOfDimension() const { return mLens.size(); }
std::size_t HostTensorDescriptor::GetElementSize() const
@@ -57,3 +45,14 @@ std::ostream& operator<<(std::ostream& os, const HostTensorDescriptor& desc)
return os;
}
std::ostream& operator<<(std::ostream& os, HostTensorDescriptor::ChosenLayout tag)
{
switch(tag)
{
case HostTensorDescriptor::ChosenLayout::Original: os << "Original"; break;
case HostTensorDescriptor::ChosenLayout::RowMajor: os << "RowMajor"; break;
case HostTensorDescriptor::ChosenLayout::ColumnMajor: os << "ColumnMajor"; break;
}
return os;
}