"git@developer.sourcefind.cn:OpenDAS/torch-cluster.git" did not exist on "9b0759b5e04cf5515e053243606e26402ba4aa96"
Commit b2c458ca authored by rocking's avatar rocking
Browse files

Add missing index

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