    def reset_dev_state(self, dev_id: int) -> None:
        """Rest the device id state to idle

        Parameters
        ----------
        dev_id : int
            The id of device will be reset
        """
        self._dev_stats[dev_id] = False
Rest the device id state to idle

Parameters
----------
dev_id : int
The id of device will be reset
python/aitemplate/backend/task_runner.py