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

implicit gemm v1r3 nchw_cyxk_nkhw

Browse Source
This commit is contained in:
Chao Liu
2019-04-25 15:14:39 -05:00
parent 569ad66e2a
commit a903146427
18 changed files with 1202 additions and 523 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -137,11 +137,16 @@ struct ConstantTensorDescriptor
}
template <index_t... Is>
__host__ __device__ static constexpr index_t Get1dIndex(Sequence<Is...> multi_id)
__host__ __device__ static constexpr index_t Get1dIndex(Sequence<Is...> /*multi_id*/)
{
static_assert(sizeof...(Is) == nDim, "wrong! Dimension not consistent");
return Get1dIndex(Is...);
constexpr auto multi_id = Sequence<Is...>{};
constexpr auto seq_tmp =
transform_sequences(mod_conv::multiplies<index_t>{}, multi_id, GetStrides());
return accumulate_on_sequence(seq_tmp, mod_conv::plus<index_t>{}, Number<0>{});
}
__host__ __device__ static Array<index_t, nDim> GetMultiIndex(index_t id)