jax.experimental.multihost_utils.sync_global_devices

jax.experimental.multihost_utils.sync_global_devices#

jax.experimental.multihost_utils.sync_global_devices(name)[source]#

在所有主机/设备之间创建一个屏障。

参数:

name (str)