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

fix build issue

Browse Source
This commit is contained in:
Chao Liu
2019-06-06 17:13:15 -05:00
parent 7a89684f92
commit 08c692433e
1 changed files with 7 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -66,10 +66,11 @@ struct ConstantTensorDescriptor
{
}
template <class X>
__host__ __device__ constexpr void operator()(X IDim) const
template <index_t IDim_>
__host__ __device__ constexpr void operator()(Number<IDim_>) const
{
constexpr auto IDim_p1 = IDim + Number<1>{};
constexpr auto IDim = Number<IDim_>{};
constexpr auto IDim_p1 = Number<IDim_ + 1>{};
is_continuous =
is_continuous && (GetStride(IDim) >= GetStride(IDim_p1) &&
@@ -178,9 +179,10 @@ struct ConstantTensorDescriptor
{
}
template <class X>
__host__ __device__ constexpr void operator()(X IDim) const
template <class IDim_>
__host__ __device__ constexpr void operator()(IDim_) const
{
constexpr auto IDim = IDim_{};
constexpr index_t stride = PackedStrides::Get(IDim);
multi_id.Set(IDim, id / stride);
id -= multi_id[IDim] * stride;