jax.experimental.multihost_utils.sync_global_devices

jax.experimental.multihost_utils.sync_global_devices#

jax.experimental.multihost_utils.sync_global_devices(name)[源代码][源代码]#

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

参数:

name (str)