Ignore operations not in the current syntax tree.

m4rs-mt · Jun 20, 2024 · 32a441a · 32a441a
1 parent db72095
commit 32a441a
Showing 1 changed file with 9 additions and 3 deletions.
diff --git a/Src/ILGPU.Analyzers/MethodUtil.cs b/Src/ILGPU.Analyzers/MethodUtil.cs
@@ -35,12 +35,18 @@ public static class MethodUtil
                 {
                     IsPartialDefinition: false,
                     DeclaringSyntaxReferences: var refs,
-                } => refs.Length > 0 ? model.GetOperation(refs[0].GetSyntax()) : null,
+                } => refs.Length > 0 ? GetOperationIfInTree(model, refs[0].GetSyntax()) : null,
                 {
                     PartialImplementationPart: { DeclaringSyntaxReferences: var refs },
-                } => refs.Length > 0 ? model.GetOperation(refs[0].GetSyntax()) : null,
+                } => refs.Length > 0 ? GetOperationIfInTree(model, refs[0].GetSyntax()) : null,
                 _ => null
             };
         }
+
+        private static IOperation? GetOperationIfInTree(SemanticModel model, SyntaxNode node)
+        {
+            var root = model.SyntaxTree.GetRoot();
+            return root.Contains(node) ? model.GetOperation(node) : null;
+        }
     }
-}
+}