wehub-resource-sync
36 lines
871 B
Python
36 lines
871 B
Python
"""
|
|
Constants.
|
|
|
|
Contains constants used to setup collective groups.
|
|
"""
|
|
import hashlib
|
|
import os
|
|
from enum import Enum, auto
|
|
|
|
|
|
def get_store_name(group_name: str):
|
|
"""Generate the unique name for the NCCLUniqueID store (named actor).
|
|
|
|
Args:
|
|
group_name: unique user name for the store.
|
|
|
|
Returns:
|
|
SHA256-hexlified name for the store.
|
|
"""
|
|
if not group_name:
|
|
raise ValueError("group_name is None.")
|
|
hexlified_name = hashlib.sha256(group_name.encode()).hexdigest()
|
|
return hexlified_name
|
|
|
|
|
|
class ENV(Enum):
|
|
"""ray.util.collective environment variables."""
|
|
|
|
NCCL_USE_MULTISTREAM = auto(), lambda v: (v or "True") == "True"
|
|
|
|
@property
|
|
def val(self):
|
|
"""Return the output of the lambda against the system's env value."""
|
|
_, default_fn = self.value
|
|
return default_fn(os.getenv(self.name))
|