Skip to content

GitLab

Commit 44b66c41 authored by rocking's avatar rocking
Browse files

Refine naming

parent 14d29856
Show whitespace changes
Inline Side-by-side
Showing with 12 additions and 12 deletions
include/ck/tensor_operation/gpu/grid/gemm_layernorm/gridwise_welford_second_half_layernorm2d.hpp
View file @ 44b66c41
......@@ -239,7 +239,7 @@ struct GridwiseWelfordSecondHalfLayernorm2d
block_work_idx[I0] * M_BlockTileSize + thread_m_cluster_id * MThreadSliceSize,
block_work_idx[I1] * N_BlockTileSize + thread_n_cluster_id * NThreadSliceSize));
auto threadwise_gamma_load_m_n =
auto threadwise_gamma_load_n =
ThreadwiseTensorSliceTransfer_v2<GammaDataType,
ComputeDataType,
decltype(gamma_grid_desc_n),
......@@ -254,7 +254,7 @@ struct GridwiseWelfordSecondHalfLayernorm2d
make_multi_index(block_work_idx[I1] * N_BlockTileSize +
thread_n_cluster_id * NThreadSliceSize));
auto threadwise_beta_load_m_n =
auto threadwise_beta_load_n =
ThreadwiseTensorSliceTransfer_v2<BetaDataType,
ComputeDataType,
decltype(beta_grid_desc_n),
......@@ -358,7 +358,7 @@ struct GridwiseWelfordSecondHalfLayernorm2d
});
});
threadwise_gamma_load_m_n.Run(gamma_grid_desc_n,
threadwise_gamma_load_n.Run(gamma_grid_desc_n,
gamma_global_val_buf,
thread_buffer_desc_n,
make_tuple(I0),
......@@ -371,7 +371,7 @@ struct GridwiseWelfordSecondHalfLayernorm2d
});
});
threadwise_beta_load_m_n.Run(beta_grid_desc_n,
threadwise_beta_load_n.Run(beta_grid_desc_n,
beta_global_val_buf,
thread_buffer_desc_n,
make_tuple(I0),
......
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