Skip to content

GitLab

"git@developer.sourcefind.cn:OpenDAS/megatron-lm.git" did not exist on "84a5997a2949fa15a7c663882cf5187f3f58fadd"
Commit 08c69243 authored by Chao Liu's avatar Chao Liu
Browse files

fix build issue

parent 7a89684f
Hide whitespace changes
Inline Side-by-side
Showing with 7 additions and 5 deletions
src/include/ConstantTensorDescriptor.hip.hpp
View file @ 08c69243
...@@ -66,10 +66,11 @@ struct ConstantTensorDescriptor ...@@ -66,10 +66,11 @@ struct ConstantTensorDescriptor
{ {
} }
template <class X> template <index_t IDim_>
__host__ __device__ constexpr void operator()(X IDim) const __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 =
is_continuous && (GetStride(IDim) >= GetStride(IDim_p1) && is_continuous && (GetStride(IDim) >= GetStride(IDim_p1) &&
...@@ -178,9 +179,10 @@ struct ConstantTensorDescriptor ...@@ -178,9 +179,10 @@ struct ConstantTensorDescriptor
{ {
} }
template <class X> template <class IDim_>
__host__ __device__ constexpr void operator()(X IDim) const __host__ __device__ constexpr void operator()(IDim_) const
{ {
constexpr auto IDim = IDim_{};
constexpr index_t stride = PackedStrides::Get(IDim); constexpr index_t stride = PackedStrides::Get(IDim);
multi_id.Set(IDim, id / stride); multi_id.Set(IDim, id / stride);
id -= multi_id[IDim] * stride; id -= multi_id[IDim] * stride;
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment