Skip to content

Fix: sink layer id and tensor write-back scheduling#96

Open
sjoks93 wants to merge 2 commits intoKULeuven-MICAS:masterfrom
sjoks93:bug-fix-branch
Open

Fix: sink layer id and tensor write-back scheduling#96
sjoks93 wants to merge 2 commits intoKULeuven-MICAS:masterfrom
sjoks93:bug-fix-branch

Conversation

@sjoks93
Copy link

@sjoks93 sjoks93 commented Feb 16, 2026

I created this fix for sink layer ids. The logic is as follows:

Current implementation looks for layer ids where ALL nodes associated with id have no successors. This is wrong, as intra-layer partitioning leads to edges between sink nodes. Instead, if ONE node with a certain id has no successor, then ALL nodes with that id should be considered as sink nodes.

EDIT: I have added a second bug fix related to scheduling tensor write-back. Without this fix, a tensor write-back may overlap with another tensor transfer (from off-chip to a core). I wanted to do this in a more proper way with two separate pull request, but I have realized my mistake too late. Hence, I combined both fixes in this pull request. (Sorry)
Please review each commit separately and decide if any or both are acceptable

@sjoks93 sjoks93 changed the title Fix: sink layer id Fix: sink layer id and tensor write-back scheduling Feb 16, 2026
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants