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

refactor

Browse Source
This commit is contained in:
Chao Liu
2019-04-01 15:17:22 -05:00
parent d058d16407
commit e43d7bc63c
13 changed files with 873 additions and 928 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
src/include/ConstantTensorDescriptor.hip.hpp
View File

@@ -137,7 +137,10 @@ struct ConstantTensorDescriptor
}
};
return static_const_reduce_n<nDim>{}(GetElementSpace_f{}, add{}) + align.Get();
index_t element_space_unaligned =
static_const_reduce_n<nDim>{}(GetElementSpace_f{}, add{}) + 1;
return align.Get() * ((element_space_unaligned + align.Get() - 1) / align.Get());
}
template <class... Is>