    def assigned_dev(self) -> int:
        """Return the assigned device id for the task

        Returns
        -------
        int
            Assigned device id
        """
        return self._assigned_dev
Return the assigned device id for the task

Returns
-------
int
Assigned device id
python/aitemplate/backend/task_runner.py