INNER CODE UNIT · Python
size
bytedance/byteps · byteps/common/__init__.py:89
if size == -1:
raise ValueError(
'BytePS has not been initialized; use bps.init().')
return size
def local_size(self):
"""A function that returns the number of BytePS processes within the
node the current process is running on.
Returns:
An integer scalar containing the number of local BytePS processes.
"""
local_size = self.C_LIB_CTYPES.byteps_local_size()
if local_size == -1:
raise ValueError(
'BytePS has not been initialized; use bps.init().')
return local_size
def rank(self):