Skip to content

Commit

Permalink
Refine variable name
Browse files Browse the repository at this point in the history
  • Loading branch information
rocking5566 committed Nov 8, 2023
1 parent 74495de commit 694408d
Show file tree
Hide file tree
Showing 3 changed files with 12 additions and 12 deletions.
6 changes: 3 additions & 3 deletions example/53_layernorm_bwd/layernorm2d_bwd_fp16.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -62,11 +62,11 @@ using GammaBetaDeviceInstance = ck::tensor_operation::device::DeviceNormalizatio
32, // ClusterReduce
8, // SliceInvarient
1, // SliceReduce
false, // IsDYSrcVectorDimReduced
false, // IsDYFastestDimReduced
8, // DYSrcVectorSize
false, // IsXSrcVectorDimReduced
false, // IsXFastestDimReduced
8, // XSrcVectorSize
true, // IsMeanInvStdSrcVectorDimReduced
true, // IsMeanInvStdFastestDimReduced
1, // MeanInvStdSrcVectorSize
1, // DGammaDstVectorSize
1>; // DBetaDstVectorSize
Expand Down
6 changes: 3 additions & 3 deletions example/54_groupnorm_bwd/groupnorm_bwd_fp16.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -56,11 +56,11 @@ using GammaBetaDeviceInstance = ck::tensor_operation::device::DeviceNormalizatio
32, // ClusterReduce
8, // SliceInvarient
1, // SliceReduce
false, // IsDYSrcVectorDimReduced
false, // IsDYFastestDimReduced
8, // DYSrcVectorSize
false, // IsXSrcVectorDimReduced
false, // IsXFastestDimReduced
8, // XSrcVectorSize
false, // IsMeanInvStdSrcVectorDimReduced
false, // IsMeanInvStdFastestDimReduced
1, // MeanInvStdSrcVectorSize
1, // DGammaDstVectorSize
1>; // DBetaDstVectorSize
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -70,11 +70,11 @@ template <typename DYDataType,
index_t KThreadClusterSize,
index_t MThreadSliceSize,
index_t KThreadSliceSize,
bool IsDYSrcVectorDimReduced,
bool IsDYFastestDimReduced,
index_t DYSrcVectorSize,
bool IsXSrcVectorDimReduced,
bool IsXFastestDimReduced,
index_t XSrcVectorSize,
bool IsMeanInvStdSrcVectorDimReduced,
bool IsMeanInvStdFastestDimReduced,
index_t MeanInvStdSrcVectorSize,
index_t DGammaDstVectorSize,
index_t DBetaDstVectorSize>
Expand All @@ -88,9 +88,9 @@ struct DeviceNormalizationBwdGammaBetaImpl
NumReduceDim>
{

static constexpr index_t DYSrcVectorDim = IsDYSrcVectorDimReduced ? 1 : 0;
static constexpr index_t XSrcVectorDim = IsXSrcVectorDimReduced ? 1 : 0;
static constexpr index_t MeanInvStdSrcVectorDim = IsMeanInvStdSrcVectorDimReduced ? 1 : 0;
static constexpr index_t DYSrcVectorDim = IsDYFastestDimReduced ? 1 : 0;
static constexpr index_t XSrcVectorDim = IsXFastestDimReduced ? 1 : 0;
static constexpr index_t MeanInvStdSrcVectorDim = IsMeanInvStdFastestDimReduced ? 1 : 0;

static_assert(BlockSize == MThreadClusterSize * KThreadClusterSize);

Expand Down

0 comments on commit 694408d

Please sign in to comment.