From c60ac41c4a465818144cf99501bd4ed234f25402 Mon Sep 17 00:00:00 2001
From: John Omotani <john.omotani@ukaea.uk>
Date: Mon, 27 Jan 2025 16:29:11 +0000
Subject: [PATCH] fix stricter array in reconcile boundaries
---
moment_kinetics/src/calculus.jl | 24 ++++++++++++------------
1 file changed, 12 insertions(+), 12 deletions(-)
diff --git a/moment_kinetics/src/calculus.jl b/moment_kinetics/src/calculus.jl
index a55d3ec0b0..4343764786 100644
--- a/moment_kinetics/src/calculus.jl
+++ b/moment_kinetics/src/calculus.jl
@@ -528,12 +528,12 @@ end
if irank == 0
if coord.bc == "periodic"
#update the extreme lower endpoint with data from irank = nrank -1
- receive_buffer1 .= 0.5*(receive_buffer1 .+ dfdx_lower_endpoints)
+ @. receive_buffer1 = 0.5*(receive_buffer1 + dfdx_lower_endpoints)
else #directly use value from Cheb
receive_buffer1 .= dfdx_lower_endpoints
end
else # enforce continuity at lower endpoint
- receive_buffer1 .= 0.5*(receive_buffer1 .+ dfdx_lower_endpoints)
+ @. receive_buffer1 = 0.5*(receive_buffer1 + dfdx_lower_endpoints)
end
#now update the df1d array -- using a slice appropriate to the dimension reconciled
assign_endpoint!(df1d,receive_buffer1,"lower",coord)
@@ -541,12 +541,12 @@ end
if irank == nrank-1
if coord.bc == "periodic"
#update the extreme upper endpoint with data from irank = 0
- receive_buffer2 .= 0.5*(receive_buffer2 .+ dfdx_upper_endpoints)
+ @. receive_buffer2 = 0.5*(receive_buffer2 + dfdx_upper_endpoints)
else #directly use value from Cheb
receive_buffer2 .= dfdx_upper_endpoints
end
else # enforce continuity at upper endpoint
- receive_buffer2 .= 0.5*(receive_buffer2 .+ dfdx_upper_endpoints)
+ @. receive_buffer2 = 0.5*(receive_buffer2 + dfdx_upper_endpoints)
end
#now update the df1d array -- using a slice appropriate to the dimension reconciled
assign_endpoint!(df1d,receive_buffer2,"upper",coord)
@@ -707,28 +707,28 @@ end
if irank == 0
if coord.bc == "periodic"
#update the extreme lower endpoint with data from irank = nrank -1
- receive_buffer1 .= 0.5*(receive_buffer1 .+ dfdx_lower_endpoints)
+ @. receive_buffer1 = 0.5*(receive_buffer1 + dfdx_lower_endpoints)
else #directly use value from Cheb
receive_buffer1 .= dfdx_lower_endpoints
end
else # enforce continuity at lower endpoint
- receive_buffer1 .= 0.5*(receive_buffer1 .+ dfdx_lower_endpoints)
+ @. receive_buffer1 = 0.5*(receive_buffer1 + dfdx_lower_endpoints)
end
#now update the df1d array -- using a slice appropriate to the dimension reconciled
- @views df1d[:,:,1] .= receive_buffer1
+ df1d[:,:,1] .= receive_buffer1
if irank == nrank-1
if coord.bc == "periodic"
#update the extreme upper endpoint with data from irank = 0
- receive_buffer2 .= 0.5*(receive_buffer2 .+ dfdx_upper_endpoints)
+ @. receive_buffer2 = 0.5*(receive_buffer2 + dfdx_upper_endpoints)
else #directly use value from Cheb
receive_buffer2 .= dfdx_upper_endpoints
end
else # enforce continuity at upper endpoint
- receive_buffer2 .= 0.5*(receive_buffer2 .+ dfdx_upper_endpoints)
+ @. receive_buffer2 = 0.5*(receive_buffer2 + dfdx_upper_endpoints)
end
#now update the df1d array -- using a slice appropriate to the dimension reconciled
- @views df1d[:,:,end] .= receive_buffer2
+ df1d[:,:,end] .= receive_buffer2
end
# synchronize buffers
@@ -796,7 +796,7 @@ end
apply_adv_fac!(receive_buffer1,adv_fac_lower_endpoints,dfdx_lower_endpoints,1)
end
#now update the df1d array -- using a slice appropriate to the dimension reconciled
- @views df1d[:,:,1] .= receive_buffer1
+ df1d[:,:,1] .= receive_buffer1
if irank == nrank-1
if coord.bc == "periodic"
@@ -809,7 +809,7 @@ end
apply_adv_fac!(receive_buffer2,adv_fac_upper_endpoints,dfdx_upper_endpoints,-1)
end
#now update the df1d array -- using a slice appropriate to the dimension reconciled
- @views df1d[:,:,end] .= receive_buffer2
+ df1d[:,:,end] .= receive_buffer2
end
# synchronize buffers