From 7800757c9aa72aeedfd20f4a04b40f4f25b8da6a Mon Sep 17 00:00:00 2001 From: Maths <644311798@qq.com> Date: Tue, 26 Mar 2024 18:21:13 +0800 Subject: [PATCH] Update svole_kernel.cc --- yacl/kernels/svole_kernel.cc | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/yacl/kernels/svole_kernel.cc b/yacl/kernels/svole_kernel.cc index 4fae0293..a92b045b 100644 --- a/yacl/kernels/svole_kernel.cc +++ b/yacl/kernels/svole_kernel.cc @@ -88,6 +88,7 @@ void SVoleKernel::eval_multithread(const std::shared_ptr& lctx, } tl_c[threads - 1] = out_c.subspan(iter_size * (threads - 1), last_size); + *out_delta = std::get(core_).GetDelta(); uint128_t shared_seed = SyncSeedSend(lctx); auto lctx_tl = SetupLink(lctx, threads); /* thread-local link */ @@ -98,7 +99,6 @@ void SVoleKernel::eval_multithread(const std::shared_ptr& lctx, auto tl_core = core_; std::get(tl_core).SetOTCounter(tl_seed); std::get(tl_core).SfSend(lctx_tl[i], tl_c[i]); - *out_delta = std::get(tl_core).GetDelta(); }; for (int i = 0; i < threads; ++i) { pool.Submit(task, i);