Commit b2c458ca authored by rocking's avatar rocking
Browse files

Add missing index

parent 69b925b9
...@@ -1025,7 +1025,7 @@ struct GridwiseGemmMultipleDWelfordFirstHalf_xdl_cshuffle ...@@ -1025,7 +1025,7 @@ struct GridwiseGemmMultipleDWelfordFirstHalf_xdl_cshuffle
static_for<0, PostShuffleThreadSliceSize_M, 1>{}([&](auto j) { static_for<0, PostShuffleThreadSliceSize_M, 1>{}([&](auto j) {
block_sync_lds(); block_sync_lds();
count_thread_buf = threadwise_welfords(i).cur_count_; count_thread_buf(j) = threadwise_welfords(i).cur_count_;
BlockwiseWelford::Run( BlockwiseWelford::Run(
mean_thread_buf(j), var_thread_buf(j), count_thread_buf(j)); mean_thread_buf(j), var_thread_buf(j), count_thread_buf(j));
}); });
......
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