From a4cb5fbca2aedcfd5a64b570f9ba48c717cf6123 Mon Sep 17 00:00:00 2001
From: Abdul Rafae Noor <arnoor2@tyler.cs.illinois.edu>
Date: Sun, 24 Jan 2021 20:31:46 -0600
Subject: [PATCH] Bugfix FuseHPVMTensorNodes: Added NULL check in
 findNextNodeInSequence

---
 .../lib/Transforms/FuseHPVMTensorNodes/FuseHPVMTensorNodes.cpp | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/hpvm/lib/Transforms/FuseHPVMTensorNodes/FuseHPVMTensorNodes.cpp b/hpvm/lib/Transforms/FuseHPVMTensorNodes/FuseHPVMTensorNodes.cpp
index e28134f09f..c46a221561 100644
--- a/hpvm/lib/Transforms/FuseHPVMTensorNodes/FuseHPVMTensorNodes.cpp
+++ b/hpvm/lib/Transforms/FuseHPVMTensorNodes/FuseHPVMTensorNodes.cpp
@@ -129,6 +129,9 @@ static DFNode *findNextNodeInSequence(DFNode *SrcN) {
     }
   }
 
+  if(!DstN)
+      return NULL;
+
   // If we reach this point, DstN is the unique successor of SrcN
 
   // Now, test that the DstN has a single predeccessor except Root (dummy)
-- 
GitLab