XLB
XLB copied to clipboard
assign_fields_sharded with pre-defined initialization
The assign_fields_sharded doesn't work correctly if we define initial rho0 and u0 on multi-GPU systems. It is a simple fix and I'll submit a PR for it. Adding it here so that I don't forget about it.