def __init__(self, devs: List[int]) -> None:
        """Initialize a Device Farm given a list of device ids.

        Parameters
        ----------
        devs : List[int]
            List of device ids in int
        """
        if isinstance(devs, int):
            devs = list(range(devs))
        assert isinstance(devs, list)
        self._dev_stats = OrderedDict()
        self._devs = devs
        for dev in devs:
            self._dev_stats[dev] = False
Initialize a Device Farm given a list of device ids.

        Parameters
        ----------
        devs : List[int]
            List of device ids in int
        
python/aitemplate/backend/task_runner.py