From 5e5484e74998332d17383c7774663c23a8122006 Mon Sep 17 00:00:00 2001 From: Francois Lanusse Date: Sun, 7 Jul 2024 20:00:22 -0400 Subject: [PATCH] Update README.md --- README.md | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/README.md b/README.md index 009e04e..733127c 100644 --- a/README.md +++ b/README.md @@ -59,7 +59,7 @@ with mesh: halo_extents=(32,32,32), halo_periods=(True,True,True)) # Remove the halo regions - recarray = jaxdecomp.slice_unpad(exchanged_reduced, padding_width, pdims) + recarray = jaxdecomp.slice_unpad(exchanged_array, padding_width, pdims) # Gather the results (only if it fits on CPU memory) gathered_array = multihost_utils.process_allgather(recarray, tiled=True)