Skip to content

GitLab

Commit 54f52881 authored by wangshaojie6's avatar wangshaojie6
Browse files

remove ignore in generating tuple of vector

parent cabe8d80
Hide whitespace changes
Inline Side-by-side
Showing with 4 additions and 8 deletions
include/ck/tensor_operation/gpu/grid/gridwise_layernorm_welford_variance.hpp
View file @ 54f52881
...@@ -126,8 +126,7 @@ struct GridwiseLayernormWelfordVariance_mk_to_mk ...@@ -126,8 +126,7 @@ struct GridwiseLayernormWelfordVariance_mk_to_mk
p_y_global, y_grid_desc_m_k.GetElementSpaceSize()); p_y_global, y_grid_desc_m_k.GetElementSpaceSize());
auto x_thread_buf = generate_tuple( auto x_thread_buf = generate_tuple(
[&](auto i) { [&](auto) {
ignore = i;
return StaticBuffer<AddressSpaceEnum::Vgpr, return StaticBuffer<AddressSpaceEnum::Vgpr,
AccDataType, AccDataType,
MThreadSliceSize * XSrcVectorSize, MThreadSliceSize * XSrcVectorSize,
...@@ -136,8 +135,7 @@ struct GridwiseLayernormWelfordVariance_mk_to_mk ...@@ -136,8 +135,7 @@ struct GridwiseLayernormWelfordVariance_mk_to_mk
Number<XThreadBufferNumber>{}); Number<XThreadBufferNumber>{});
auto gamma_thread_buf = generate_tuple( auto gamma_thread_buf = generate_tuple(
[&](auto i) { [&](auto) {
ignore = i;
return StaticBuffer<AddressSpaceEnum::Vgpr, return StaticBuffer<AddressSpaceEnum::Vgpr,
AccDataType, AccDataType,
MThreadSliceSize * GammaSrcVectorSize, MThreadSliceSize * GammaSrcVectorSize,
...@@ -146,8 +144,7 @@ struct GridwiseLayernormWelfordVariance_mk_to_mk ...@@ -146,8 +144,7 @@ struct GridwiseLayernormWelfordVariance_mk_to_mk
Number<GammaThreadBufferNumber>{}); Number<GammaThreadBufferNumber>{});
auto beta_thread_buf = generate_tuple( auto beta_thread_buf = generate_tuple(
[&](auto i) { [&](auto) {
ignore = i;
return StaticBuffer<AddressSpaceEnum::Vgpr, return StaticBuffer<AddressSpaceEnum::Vgpr,
AccDataType, AccDataType,
MThreadSliceSize * BetaSrcVectorSize, MThreadSliceSize * BetaSrcVectorSize,
...@@ -156,8 +153,7 @@ struct GridwiseLayernormWelfordVariance_mk_to_mk ...@@ -156,8 +153,7 @@ struct GridwiseLayernormWelfordVariance_mk_to_mk
Number<BetaThreadBufferNumber>{}); Number<BetaThreadBufferNumber>{});
auto y_thread_buf = generate_tuple( auto y_thread_buf = generate_tuple(
[&](auto i) { [&](auto) {
ignore = i;
return StaticBuffer<AddressSpaceEnum::Vgpr, return StaticBuffer<AddressSpaceEnum::Vgpr,
AccDataType, AccDataType,
MThreadSliceSize * YDstVectorSize, MThreadSliceSize * YDstVectorSize,
......
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