[flang][openacc] Add routine info attribute to the function

[clementval]

Summary

Attch the routine info attribute to the func operation.

[razvanlupusoru]

Looks good!

flang/lib/Lower/OpenACC.cpp

genACC(Fortran::lower::AbstractConverter &converter,
   mlir::ModuleOp mod = builder.getModule();
   mlir::OpBuilder modBuilder(mod.getBodyRegion());
   std::stringstream routineOpName;
   routineOpName << accRoutinePrefix.str() << routineCounter++;
   modBuilder.create<mlir::acc::RoutineOp>(
       loc, routineOpName.str(), func.getName(), mlir::StringAttr{},
       mlir::UnitAttr{}, mlir::UnitAttr{}, mlir::UnitAttr{}, mlir::UnitAttr{},
       mlir::UnitAttr{}, mlir::UnitAttr{}, mlir::IntegerAttr{});
+
+  llvm::SmallVector<mlir::SymbolRefAttr> routines;
+  if (func.getOperation()->hasAttr(mlir::acc::getRoutineInfoAttrName())) {
+    auto routineInfo =
+        func.getOperation()->getAttrOfType<mlir::acc::RoutineInfoAttr>(
+            mlir::acc::getRoutineInfoAttrName());
+    routines.append(routineInfo.getAccRoutines().begin(),
+                    routineInfo.getAccRoutines().end());
+  }
+  routines.push_back(builder.getSymbolRefAttr(routineOpName.str()));
+  func.getOperation()->setAttr(
+      mlir::acc::getRoutineInfoAttrName(),
+      mlir::acc::RoutineInfoAttr::get(builder.getContext(), routines));
 }
 
 void Fortran::lower::genOpenACCConstruct(
     Fortran::lower::AbstractConverter &converter,
     Fortran::semantics::SemanticsContext &semanticsContext,
     Fortran::lower::pft::Evaluation &eval,
     const Fortran::parser::OpenACCConstruct &accConstruct) {
 

flang/test/Lower/OpenACC/acc-routine.f90

 ! This test checks lowering of OpenACC routine directive.
 
 ! RUN: bbc -fopenacc -emit-fir %s -o - | FileCheck %s
 
 subroutine acc_routine1()
   !$acc routine
 end subroutine
 
 ! CHECK: acc.routine @acc_routine_0 func(@_QPacc_routine1)
-! CHECK-LABEL: func.func @_QPacc_routine1()
+! CHECK-LABEL: func.func @_QPacc_routine1() attributes {acc.routine_info = #acc.routine_info<[@acc_routine_0]>}

mlir/include/mlir/Dialect/OpenACC/OpenACC.h

 static constexpr StringLiteral getDeclareAttrName() {
   return StringLiteral("acc.declare");
 }
 
 static constexpr StringLiteral getDeclareActionAttrName() {
   return StringLiteral("acc.declare_action");
 }
 
+static constexpr StringLiteral getRoutineInfoAttrName() {
+  return StringLiteral("acc.routine_info");
+}
+
 } // namespace acc
 } // namespace mlir
 
 #endif // MLIR_DIALECT_OPENACC_OPENACC_H_