ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-06-08 15:30:23 +00:00
Code Issues Packages Projects Releases Wiki Activity

Transpose -> transpose

Browse Source
This commit is contained in:
danyao12
2024-05-29 16:54:26 +08:00
parent 58f61716b5
commit 87f73f30e8
2 changed files with 10 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -481,7 +481,7 @@ struct HostTensor
return mData[mDesc.GetOffsetFromMultiIndex(idx)];
}
HostTensor<T> Transpose(std::vector<size_t> axes = {}) const
HostTensor<T> transpose(std::vector<size_t> axes = {}) const
{
if(axes.empty())
{
@@ -491,7 +491,7 @@ struct HostTensor
if(axes.size() != mDesc.get_num_of_dimension())
{
throw std::runtime_error(
"HostTensor::Transpose(): size of axes must match tensor dimension");
"HostTensor::transpose(): size of axes must match tensor dimension");
}
std::vector<size_t> tlengths, tstrides;
for(const auto& axis : axes)
@@ -504,9 +504,9 @@ struct HostTensor
return ret;
}
HostTensor<T> Transpose(std::vector<size_t> axes = {})
HostTensor<T> transpose(std::vector<size_t> axes = {})
{
return const_cast<HostTensor<T> const*>(this)->Transpose(axes);
return const_cast<HostTensor<T> const*>(this)->transpose(axes);
}
typename Data::iterator begin() { return mData.begin(); }