Add more debug info to DaCe (pass SourceLocation from past/foast to i…

…tir, and from itir to the SDFG): Preserve Location through Visitors [WIP]

src/gt4py/eve/visitors.py

@@ -21,8 +21,6 @@
 import copy
 from typing import ClassVar
 
-from gt4py.next.ffront.field_operator_ast import Name
-
 from . import concepts, trees
 from .extended_typing import Any
 from .type_definitions import NOTHING
@@ -203,6 +201,6 @@ def generic_visit(self, node: concepts.RootNode, **kwargs: Any) -> Any:
 class PreserveLocation(NodeVisitor):
     def visit(self, node: concepts.RootNode, **kwargs: Any) -> Any:
         result = super().visit(node, **kwargs)
-        if hasattr(node, "location") and hasattr(result, "location") and not isinstance(node, Name):
+        if hasattr(node, "location") and hasattr(result, "location"):
             result.location = node.location
         return result

src/gt4py/next/ffront/foast_to_itir.py

@@ -15,9 +15,8 @@
 import dataclasses
 from typing import Any, Callable, Optional
 
-from gt4py.eve import NodeTranslator
+from gt4py.eve import NodeTranslator, concepts, extended_typing
 from gt4py.eve.utils import UIDGenerator
-from gt4py.eve.visitors import PreserveLocation
 from gt4py.next.ffront import (
     dialect_ast_enums,
     fbuiltins,
@@ -40,7 +39,7 @@ def promote_to_list(
 
 
 @dataclasses.dataclass
-class FieldOperatorLowering(PreserveLocation, NodeTranslator):
+class FieldOperatorLowering(NodeTranslator):
     """
     Lower FieldOperator AST (FOAST) to Iterator IR (ITIR).
 
@@ -72,6 +71,16 @@ class FieldOperatorLowering(PreserveLocation, NodeTranslator):
     def apply(cls, node: foast.LocatedNode) -> itir.Expr:
         return cls().visit(node)
 
+    def visit(self, node: concepts.RootNode, **kwargs: extended_typing.Any) -> extended_typing.Any:
+        result = super().visit(node, **kwargs)
+        if (
+            hasattr(node, "location")
+            and hasattr(result, "location")
+            and not isinstance(node, foast.Name)
+        ):
+            result.location = node.location
+        return result
+
     def visit_FunctionDefinition(
         self, node: foast.FunctionDefinition, **kwargs
     ) -> itir.FunctionDefinition: