Share the VGPR of x and y

f61c37d0 · rocking · 491c0631 · f61c37d0
Commit f61c37d0 authored Feb 13, 2023 by rocking
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 8 deletions

include/ck/tensor_operation/gpu/grid/gridwise_normalization_welford_variance.hpp ...tion/gpu/grid/gridwise_normalization_welford_variance.hpp +1 -8

No files found.
--- a/include/ck/tensor_operation/gpu/grid/gridwise_normalization_welford_variance.hpp
+++ b/include/ck/tensor_operation/gpu/grid/gridwise_normalization_welford_variance.hpp
@@ -152,14 +152,7 @@ struct GridwiseNormalizationWelfordVariance_mk_to_mk
            },
            Number<ThreadBufferNumber>{});
-        auto y_thread_buf = generate_tuple(
+        auto& y_thread_buf = x_thread_buf;
-            [&](auto) {
-                return StaticBuffer<AddressSpaceEnum::Vgpr,
-                                    ComputeDataType,
-                                    MThreadSliceSize * YDstVectorSize,
-                                    true>{};
-            },
-            Number<ThreadBufferNumber>{});
        StaticBuffer<AddressSpaceEnum::Vgpr, ComputeDataType, MThreadSliceSize, true>
            mean_thread_buf;