[SandboxVec][DAG] Fix trim schedule

vporpo · vporpo · commit b178c2d63e07 · 2025-01-25T09:42:14.000-08:00
Fix trimSchedule by skipping instructions without a DAG Node.
diff --git a/llvm/lib/Transforms/Vectorize/SandboxVectorizer/Scheduler.cpp b/llvm/lib/Transforms/Vectorize/SandboxVectorizer/Scheduler.cpp
@@ -172,6 +172,8 @@ void Scheduler::trimSchedule(ArrayRef<Instruction *> Instrs) {
   for (auto *I = LowestI, *E = TopI->getPrevNode(); I != E;
        I = I->getPrevNode()) {
     auto *N = DAG.getNode(I);
+    if (N == nullptr)
+      continue;
     if (auto *SB = N->getSchedBundle())
       eraseBundle(SB);
   }

Original file line number	Diff line number	Diff line change
`@@ -172,6 +172,8 @@ void Scheduler::trimSchedule(ArrayRef<Instruction *> Instrs) {`
`172`	`172`	`for (auto I = LowestI, E = TopI->getPrevNode(); I != E;`
`173`	`173`	`I = I->getPrevNode()) {`
`174`	`174`	`auto *N = DAG.getNode(I);`
	`175`	`+ if (N == nullptr)`
	`176`	`+ continue;`
`175`	`177`	`if (auto *SB = N->getSchedBundle())`
`176`	`178`	`eraseBundle(SB);`
`177`	`179`	`}`