Skip to content

Commit

Permalink
Add sync after copy_from_global, don't know why works
Browse files Browse the repository at this point in the history
  • Loading branch information
sunnycase committed Nov 27, 2024
1 parent fbaf20f commit 0ec0c71
Showing 1 changed file with 1 addition and 0 deletions.
1 change: 1 addition & 0 deletions ntt/include/nncase/ntt/kernels/reshard.h
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down

0 comments on commit 0ec0c71

Please sign in to comment.