jax.experimental.multihost_utils.sync_global_devices

jax.experimental.multihost_utils.sync_global_devices(name)

在所有主機/裝置之間建立屏障。

參數:

name (str)