assign_fields_sharded with pre-defined initialization

Autodesk / XLB

XLB: Accelerated Lattice Boltzmann (XLB) for Physics-based ML

Other

238 stars 25 forks source link

assign_fields_sharded with pre-defined initialization #21

Closed mehdiataei closed 1 month ago

mehdiataei commented 1 year ago

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.

mehdiataei commented 1 month ago

This is fixed in major-refactoring