[FRONTEND] Attach lineno information to AugAssign nodes (triton-lang#8703)

visit_AugAssign works by creating fake AST nodes for the binary
operation and the assignment and visiting those. Prior to this commit,
the nodes lack the correct lineno and col_offset so any errors raised
here don't include the helpful '^' pointer to the offending line of
code; this makes debugging painful.

Author: apgoucher

python/triton/compiler/code_generator.py

@@ -703,6 +703,11 @@ def visit_AugAssign(self, node):
         lhs.ctx = ast.Load()
         rhs = ast.BinOp(lhs, node.op, node.value)
         assign = ast.Assign(targets=[node.target], value=rhs)
+        for x in ['lineno', 'col_offset', 'end_lineno', 'end_col_offset']:
+            if hasattr(node, x):
+                y = getattr(node, x)
+                setattr(rhs, x, y)
+                setattr(assign, x, y)
         self.visit(assign)
         return self.visit(lhs)