tuning_component.dataloader_affinity

worker_affinity_init_fn

def worker_affinity_init_fn(worker_id: int,
                            available_cpus: int = 0,
                            init_fn: Optional[Callable] = None)

Initializes CPU affinity for a worker process.

Arguments:

• worker_id - The ID of the worker process.
• available_cpus - The number of available CPUs.
• init_fn - An optional initialization function.