From 0ec0c71b9c38720d23a386ba08676c326b02475a Mon Sep 17 00:00:00 2001
From: sunnycase <sunnycase@live.cn>
Date: Wed, 27 Nov 2024 09:39:45 +0000
Subject: [PATCH] Add sync after copy_from_global, don't know why works

---
 ntt/include/nncase/ntt/kernels/reshard.h | 1 +
 1 file changed, 1 insertion(+)

diff --git a/ntt/include/nncase/ntt/kernels/reshard.h b/ntt/include/nncase/ntt/kernels/reshard.h
index 0cc62f939..c5a15d05a 100644
--- a/ntt/include/nncase/ntt/kernels/reshard.h
+++ b/ntt/include/nncase/ntt/kernels/reshard.h
@@ -114,6 +114,7 @@ struct reshard_impl<SrcTensor, DestTensor> {
                     typename DestTensor::global_shape_type>
             global_tensor(global_buffer);
         tensor_copy(global_tensor.view(global_offset, local.shape()), local);
+        distributed::topology_synchronize();
     }
 };
 } // namespace detail