The _parse_task_graph function currently uses a regex that only matches PTX -- ... lines, but fails to parse valid logs for other device types such as [Apple] or potentially Inte CPU/GPU, etc.
As a workaround, this patch seems to be working:
git apply REGEX_ISSUE.patch
The
_parse_task_graphfunction currently uses a regex that only matchesPTX -- ...lines, but fails to parse valid logs for other device types such as[Apple]or potentiallyInte CPU/GPU, etc.As a workaround, this patch seems to be working: