[flang] Add lowering for integer constant

[clementval]

Summary

This patch enables the lowering of integer constant.

The ScalarExprLowering class is introduced in ConvertExpr.cpp to help
the lowering of expression. This patch adds all the placeholder as well for future
expression lowering with the appropriate TODOs.

Integer constant expression are lowered to arith.constant with an integer type corresponding to the kind value.

This patch is in support of D118787

This patch is part of the upstreaming effort from fir-dev branch.

[clementval]

flang/lib/Lower/ConvertExpr.cpp

15

Sorry for the commented includes. I'll update the patch to remove them

[clementval]

Remove commented includes

[kiranchandramohan]

Thanks, @clementval. Great job in factoring out the scalar integer expressions.

Could you add to the commit message that genExprValue is the top-level function that is used in this patch? Will be a good starting point for reviewing.

LGTM.

flang/lib/Lower/ConvertExpr.cpp

252

Nit: Can the auto be removed here?
std::optional<Scalar<Type<TypeCategory::Character, KIND>>> ?

309

Nit: Is this used in this or the related patches? If not I think it might be better to skip in this one.

flang/lib/Lower/ConvertType.cpp

499

Nit: Should this be "non-INTEGER type translation not implemented"?

[clementval]

flang/lib/Lower/ConvertType.cpp

499

It should only be called for Integer type but the kind might be off. I updated the message.

[clementval]

Address review comments

[clementval]

Rebase

[clementval]

Fix clang-format

flang/include/flang/Lower/ConvertExpr.h

+//===-- Lower/ConvertExpr.h -- lowering of expressions ----------*- C++ -*-===//
+//
+// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
+// See https://llvm.org/LICENSE.txt for license information.
+// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
+//
+//===----------------------------------------------------------------------===//
+//
+// Coding style: https://mlir.llvm.org/getting_started/DeveloperGuide/
+//
+//===----------------------------------------------------------------------===//
+///
+/// Implements the conversion from Fortran::evaluate::Expr trees to FIR.
+///
+//===----------------------------------------------------------------------===//
+
+#ifndef FORTRAN_LOWER_CONVERTEXPR_H
+#define FORTRAN_LOWER_CONVERTEXPR_H
+
+#include "flang/Lower/Support/Utils.h"
+#include "flang/Optimizer/Builder/BoxValue.h"
+#include "flang/Optimizer/Builder/FIRBuilder.h"
+
+namespace mlir {
+class Location;
+}
+
+namespace Fortran::evaluate {
+template <typename>
+class Expr;
+struct SomeType;
+} // namespace Fortran::evaluate
+
+namespace Fortran::lower {
+
+class AbstractConverter;
+class SymMap;
+using SomeExpr = Fortran::evaluate::Expr<Fortran::evaluate::SomeType>;
+
+/// Create an extended expression value.
+fir::ExtendedValue createSomeExtendedExpression(mlir::Location loc,
+                                                AbstractConverter &converter,
+                                                const SomeExpr &expr,
+                                                SymMap &symMap);
+
+} // namespace Fortran::lower
+
+#endif // FORTRAN_LOWER_CONVERTEXPR_H

flang/include/flang/Lower/ConvertType.h

 namespace pft {
 struct Variable;
 }
 
 using SomeExpr = evaluate::Expr<evaluate::SomeType>;
 using SymbolRef = common::Reference<const semantics::Symbol>;
 
 /// Get a FIR type based on a category and kind.
-mlir::Type getFIRType(mlir::MLIRContext *ctxt,
-                      common::IntrinsicTypeDefaultKinds const &defaults,
-                      common::TypeCategory tc, int kind);
+mlir::Type getFIRType(mlir::MLIRContext *ctxt, common::TypeCategory tc,
+                      int kind);
 
 /// Get a FIR type based on a category.
 mlir::Type getFIRType(mlir::MLIRContext *ctxt,
                       common::IntrinsicTypeDefaultKinds const &defaults,
                       common::TypeCategory tc);
 
 /// Translate a Fortran::evaluate::DataRef to an mlir::Type.
 mlir::Type
 translateDataRefToFIRType(mlir::MLIRContext *ctxt,
                           common::IntrinsicTypeDefaultKinds const &defaults,
                           const evaluate::DataRef &dataRef);
 
-/// Translate a Fortran::evaluate::Designator<> to an mlir::Type.
-template <common::TypeCategory TC, int KIND>
-inline mlir::Type translateDesignatorToFIRType(
-    mlir::MLIRContext *ctxt, common::IntrinsicTypeDefaultKinds const &defaults,
-    const evaluate::Designator<evaluate::Type<TC, KIND>> &) {
-  return getFIRType(ctxt, defaults, TC, KIND);
-}
-
-/// Translate a Fortran::evaluate::Designator<> to an mlir::Type.
-template <common::TypeCategory TC>
-inline mlir::Type translateDesignatorToFIRType(
-    mlir::MLIRContext *ctxt, common::IntrinsicTypeDefaultKinds const &defaults,
-    const evaluate::Designator<evaluate::SomeKind<TC>> &) {
-  return getFIRType(ctxt, defaults, TC);
-}
-
 /// Translate a SomeExpr to an mlir::Type.
 mlir::Type
 translateSomeExprToFIRType(mlir::MLIRContext *ctxt,
                            common::IntrinsicTypeDefaultKinds const &defaults,
                            const SomeExpr *expr);
 
 /// Translate a Fortran::semantics::Symbol to an mlir::Type.
 mlir::Type

flang/lib/Lower/Bridge.cpp

 //===-- Bridge.cpp -- bridge to lower to MLIR -----------------------------===//
 //
 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
 // See https://llvm.org/LICENSE.txt for license information.
 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
 //
 //===----------------------------------------------------------------------===//
 //
 // Coding style: https://mlir.llvm.org/getting_started/DeveloperGuide/
 //
 //===----------------------------------------------------------------------===//
 
 #include "flang/Lower/Bridge.h"
 #include "flang/Evaluate/tools.h"
 #include "flang/Lower/CallInterface.h"
+#include "flang/Lower/ConvertExpr.h"
+#include "flang/Lower/ConvertType.h"
 #include "flang/Lower/Mangler.h"
 #include "flang/Lower/PFTBuilder.h"
 #include "flang/Lower/Runtime.h"
 #include "flang/Lower/SymbolMap.h"
 #include "flang/Lower/Todo.h"
 #include "flang/Optimizer/Support/FIRContext.h"
 #include "mlir/IR/PatternMatch.h"
 #include "mlir/Transforms/RegionUtils.h"
public:
   //===--------------------------------------------------------------------===//
 
   mlir::Value getSymbolAddress(Fortran::lower::SymbolRef sym) override final {
     return lookupSymbol(sym).getAddr();
   }
 
   fir::ExtendedValue genExprAddr(const Fortran::lower::SomeExpr &expr,
                                  mlir::Location *loc = nullptr) override final {
-    TODO_NOLOC("Not implemented. Needed for more complex expression lowering");
+    TODO_NOLOC("Not implemented genExprAddr. Needed for more complex "
+               "expression lowering");
   }
   fir::ExtendedValue
   genExprValue(const Fortran::lower::SomeExpr &expr,
                mlir::Location *loc = nullptr) override final {
-    TODO_NOLOC("Not implemented. Needed for more complex expression lowering");
+    return createSomeExtendedExpression(loc ? *loc : toLocation(), *this, expr,
+                                        localSymbols);
   }
 
   Fortran::evaluate::FoldingContext &getFoldingContext() override final {
     return foldingContext;
   }
 
   mlir::Type genType(const Fortran::evaluate::DataRef &) override final {
-    TODO_NOLOC("Not implemented. Needed for more complex expression lowering");
+    TODO_NOLOC("Not implemented genType DataRef. Needed for more complex "
+               "expression lowering");
   }
   mlir::Type genType(const Fortran::lower::SomeExpr &) override final {
-    TODO_NOLOC("Not implemented. Needed for more complex expression lowering");
+    TODO_NOLOC("Not implemented genType SomeExpr. Needed for more complex "
+               "expression lowering");
   }
   mlir::Type genType(Fortran::lower::SymbolRef) override final {
-    TODO_NOLOC("Not implemented. Needed for more complex expression lowering");
+    TODO_NOLOC("Not implemented genType SymbolRef. Needed for more complex "
+               "expression lowering");
   }
   mlir::Type genType(Fortran::common::TypeCategory tc) override final {
-    TODO_NOLOC("Not implemented. Needed for more complex expression lowering");
+    TODO_NOLOC("Not implemented genType TypeCategory. Needed for more complex "
+               "expression lowering");
   }
   mlir::Type genType(Fortran::common::TypeCategory tc,
                      int kind) override final {
-    TODO_NOLOC("Not implemented. Needed for more complex expression lowering");
+    return Fortran::lower::getFIRType(&getMLIRContext(), tc, kind);
   }
   mlir::Type genType(const Fortran::lower::pft::Variable &) override final {
-    TODO_NOLOC("Not implemented. Needed for more complex expression lowering");
+    TODO_NOLOC("Not implemented genType Variable. Needed for more complex "
+               "expression lowering");
   }
 
   void setCurrentPosition(const Fortran::parser::CharBlock &position) {
     if (position != Fortran::parser::CharBlock{})
       currentPosition = position;
   }
 
   //===--------------------------------------------------------------------===//

flang/lib/Lower/CMakeLists.txt

 get_property(dialect_libs GLOBAL PROPERTY MLIR_DIALECT_LIBS)
 
 add_flang_library(FortranLower
   Bridge.cpp
   CallInterface.cpp
   Coarray.cpp
+  ConvertExpr.cpp
   ConvertType.cpp
   Mangler.cpp
   OpenACC.cpp
   OpenMP.cpp
   PFTBuilder.cpp
   Runtime.cpp
   SymbolMap.cpp
 

flang/lib/Lower/ConvertExpr.cpp

+//===-- ConvertExpr.cpp ---------------------------------------------------===//
+//
+// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
+// See https://llvm.org/LICENSE.txt for license information.
+// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
+//
+//===----------------------------------------------------------------------===//
+//
+// Coding style: https://mlir.llvm.org/getting_started/DeveloperGuide/
+//
+//===----------------------------------------------------------------------===//
+
+#include "flang/Lower/ConvertExpr.h"
+#include "flang/Evaluate/fold.h"
+#include "flang/Evaluate/real.h"
+#include "flang/Evaluate/traverse.h"
+#include "flang/Lower/AbstractConverter.h"
+#include "flang/Lower/Todo.h"
+#include "flang/Semantics/expression.h"
+#include "flang/Semantics/symbol.h"
+#include "flang/Semantics/tools.h"
+#include "flang/Semantics/type.h"
+#include "mlir/Dialect/StandardOps/IR/Ops.h"
+#include "llvm/Support/Debug.h"
+
+#define DEBUG_TYPE "flang-lower-expr"
+
+//===----------------------------------------------------------------------===//
+// The composition and structure of Fortran::evaluate::Expr is defined in
+// the various header files in include/flang/Evaluate. You are referred
+// there for more information on these data structures. Generally speaking,
+// these data structures are a strongly typed family of abstract data types
+// that, composed as trees, describe the syntax of Fortran expressions.
+//
+// This part of the bridge can traverse these tree structures and lower them
+// to the correct FIR representation in SSA form.
+//===----------------------------------------------------------------------===//
+
+namespace {
+
+/// Lowering of Fortran::evaluate::Expr<T> expressions
+class ScalarExprLowering {
+public:
+  using ExtValue = fir::ExtendedValue;
+
+  explicit ScalarExprLowering(mlir::Location loc,
+                              Fortran::lower::AbstractConverter &converter,
+                              Fortran::lower::SymMap &symMap)
+      : location{loc}, converter{converter},
+        builder{converter.getFirOpBuilder()}, symMap{symMap} {}
+
+  mlir::Location getLoc() { return location; }
+
+  /// Generate an integral constant of `value`
+  template <int KIND>
+  mlir::Value genIntegerConstant(mlir::MLIRContext *context,
+                                 std::int64_t value) {
+    mlir::Type type =
+        converter.genType(Fortran::common::TypeCategory::Integer, KIND);
+    return builder.createIntegerConstant(getLoc(), type, value);
+  }
+
+  ExtValue genval(Fortran::semantics::SymbolRef sym) {
+    TODO(getLoc(), "genval SymbolRef");
+  }
+
+  ExtValue genval(const Fortran::evaluate::BOZLiteralConstant &) {
+    TODO(getLoc(), "genval BOZ");
+  }
+
+  /// Return indirection to function designated in ProcedureDesignator.
+  /// The type of the function indirection is not guaranteed to match the one
+  /// of the ProcedureDesignator due to Fortran implicit typing rules.
+  ExtValue genval(const Fortran::evaluate::ProcedureDesignator &proc) {
+    TODO(getLoc(), "genval ProcedureDesignator");
+  }
+
+  ExtValue genval(const Fortran::evaluate::NullPointer &) {
+    TODO(getLoc(), "genval NullPointer");
+  }
+
+  ExtValue genval(const Fortran::evaluate::StructureConstructor &ctor) {
+    TODO(getLoc(), "genval StructureConstructor");
+  }
+
+  /// Lowering of an <i>ac-do-variable</i>, which is not a Symbol.
+  ExtValue genval(const Fortran::evaluate::ImpliedDoIndex &var) {
+    TODO(getLoc(), "genval ImpliedDoIndex");
+  }
+
+  ExtValue genval(const Fortran::evaluate::DescriptorInquiry &desc) {
+    TODO(getLoc(), "genval DescriptorInquiry");
+  }
+
+  ExtValue genval(const Fortran::evaluate::TypeParamInquiry &) {
+    TODO(getLoc(), "genval TypeParamInquiry");
+  }
+
+  template <int KIND>
+  ExtValue genval(const Fortran::evaluate::ComplexComponent<KIND> &part) {
+    TODO(getLoc(), "genval ComplexComponent");
+  }
+
+  template <int KIND>
+  ExtValue genval(const Fortran::evaluate::Negate<Fortran::evaluate::Type<
+                      Fortran::common::TypeCategory::Integer, KIND>> &op) {
+    TODO(getLoc(), "genval Negate integer");
+  }
+
+  template <int KIND>
+  ExtValue genval(const Fortran::evaluate::Negate<Fortran::evaluate::Type<
+                      Fortran::common::TypeCategory::Real, KIND>> &op) {
+    TODO(getLoc(), "genval Negate real");
+  }
+  template <int KIND>
+  ExtValue genval(const Fortran::evaluate::Negate<Fortran::evaluate::Type<
+                      Fortran::common::TypeCategory::Complex, KIND>> &op) {
+    TODO(getLoc(), "genval Negate complex");
+  }
+
+#undef GENBIN
+#define GENBIN(GenBinEvOp, GenBinTyCat, GenBinFirOp)                           \
+  template <int KIND>                                                          \
+  ExtValue genval(const Fortran::evaluate::GenBinEvOp<Fortran::evaluate::Type< \
+                      Fortran::common::TypeCategory::GenBinTyCat, KIND>> &x) { \
+    TODO(getLoc(), "genval GenBinEvOp");                                       \
+  }
+
+  GENBIN(Add, Integer, mlir::arith::AddIOp)
+  GENBIN(Add, Real, mlir::arith::AddFOp)
+  GENBIN(Add, Complex, fir::AddcOp)
+  GENBIN(Subtract, Integer, mlir::arith::SubIOp)
+  GENBIN(Subtract, Real, mlir::arith::SubFOp)
+  GENBIN(Subtract, Complex, fir::SubcOp)
+  GENBIN(Multiply, Integer, mlir::arith::MulIOp)
+  GENBIN(Multiply, Real, mlir::arith::MulFOp)
+  GENBIN(Multiply, Complex, fir::MulcOp)
+  GENBIN(Divide, Integer, mlir::arith::DivSIOp)
+  GENBIN(Divide, Real, mlir::arith::DivFOp)
+  GENBIN(Divide, Complex, fir::DivcOp)
+
+  template <Fortran::common::TypeCategory TC, int KIND>
+  ExtValue genval(
+      const Fortran::evaluate::Power<Fortran::evaluate::Type<TC, KIND>> &op) {
+    TODO(getLoc(), "genval Power");
+  }
+
+  template <Fortran::common::TypeCategory TC, int KIND>
+  ExtValue genval(
+      const Fortran::evaluate::RealToIntPower<Fortran::evaluate::Type<TC, KIND>>
+          &op) {
+    TODO(getLoc(), "genval RealToInt");
+  }
+
+  template <int KIND>
+  ExtValue genval(const Fortran::evaluate::ComplexConstructor<KIND> &op) {
+    TODO(getLoc(), "genval ComplexConstructor");
+  }
+
+  template <int KIND>
+  ExtValue genval(const Fortran::evaluate::Concat<KIND> &op) {
+    TODO(getLoc(), "genval Concat<KIND>");
+  }
+
+  /// MIN and MAX operations
+  template <Fortran::common::TypeCategory TC, int KIND>
+  ExtValue
+  genval(const Fortran::evaluate::Extremum<Fortran::evaluate::Type<TC, KIND>>
+             &op) {
+    TODO(getLoc(), "genval Extremum<TC, KIND>");
+  }
+
+  template <int KIND>
+  ExtValue genval(const Fortran::evaluate::SetLength<KIND> &x) {
+    TODO(getLoc(), "genval SetLength<KIND>");
+  }
+
+  template <int KIND>
+  ExtValue genval(const Fortran::evaluate::Relational<Fortran::evaluate::Type<
+                      Fortran::common::TypeCategory::Integer, KIND>> &op) {
+    TODO(getLoc(), "genval integer comparison");
+  }
+  template <int KIND>
+  ExtValue genval(const Fortran::evaluate::Relational<Fortran::evaluate::Type<
+                      Fortran::common::TypeCategory::Real, KIND>> &op) {
+    TODO(getLoc(), "genval real comparison");
+  }
+  template <int KIND>
+  ExtValue genval(const Fortran::evaluate::Relational<Fortran::evaluate::Type<
+                      Fortran::common::TypeCategory::Complex, KIND>> &op) {
+    TODO(getLoc(), "genval complex comparison");
+  }
+  template <int KIND>
+  ExtValue genval(const Fortran::evaluate::Relational<Fortran::evaluate::Type<
+                      Fortran::common::TypeCategory::Character, KIND>> &op) {
+    TODO(getLoc(), "genval char comparison");
+  }
+
+  ExtValue
+  genval(const Fortran::evaluate::Relational<Fortran::evaluate::SomeType> &op) {
+    TODO(getLoc(), "genval comparison");
+  }
+
+  template <Fortran::common::TypeCategory TC1, int KIND,
+            Fortran::common::TypeCategory TC2>
+  ExtValue
+  genval(const Fortran::evaluate::Convert<Fortran::evaluate::Type<TC1, KIND>,
+                                          TC2> &convert) {
+    TODO(getLoc(), "genval convert<TC1, KIND, TC2>");
+  }
+
+  template <typename A>
+  ExtValue genval(const Fortran::evaluate::Parentheses<A> &op) {
+    TODO(getLoc(), "genval parentheses<A>");
+  }
+
+  template <int KIND>
+  ExtValue genval(const Fortran::evaluate::Not<KIND> &op) {
+    TODO(getLoc(), "genval Not<KIND>");
+  }
+
+  template <int KIND>
+  ExtValue genval(const Fortran::evaluate::LogicalOperation<KIND> &op) {
+    TODO(getLoc(), "genval LogicalOperation<KIND>");
+  }
+
+  /// Convert a scalar literal constant to IR.
+  template <Fortran::common::TypeCategory TC, int KIND>
+  ExtValue genScalarLit(
+      const Fortran::evaluate::Scalar<Fortran::evaluate::Type<TC, KIND>>
+          &value) {
+    if constexpr (TC == Fortran::common::TypeCategory::Integer) {
+      return genIntegerConstant<KIND>(builder.getContext(), value.ToInt64());
+    } else if constexpr (TC == Fortran::common::TypeCategory::Logical) {
+      TODO(getLoc(), "genval bool constant");
+    } else if constexpr (TC == Fortran::common::TypeCategory::Real) {
+      TODO(getLoc(), "genval real constant");
+    } else if constexpr (TC == Fortran::common::TypeCategory::Complex) {
+      TODO(getLoc(), "genval complex constant");
+    } else /*constexpr*/ {
+      llvm_unreachable("unhandled constant");
+    }
+  }
+
+  template <Fortran::common::TypeCategory TC, int KIND>
+  ExtValue
+  genval(const Fortran::evaluate::Constant<Fortran::evaluate::Type<TC, KIND>>
+             &con) {
+    if (con.Rank() > 0)
+      TODO(getLoc(), "genval array constant");
+    std::optional<Fortran::evaluate::Scalar<Fortran::evaluate::Type<TC, KIND>>>
+        opt = con.GetScalarValue();
+    assert(opt.has_value() && "constant has no value");
+    if constexpr (TC == Fortran::common::TypeCategory::Character) {
+      TODO(getLoc(), "genval char constant");
+    } else {
+      return genScalarLit<TC, KIND>(opt.value());
+    }
+  }
+
+  fir::ExtendedValue genval(
+      const Fortran::evaluate::Constant<Fortran::evaluate::SomeDerived> &con) {
+    TODO(getLoc(), "genval constant derived");
+  }
+
+  template <typename A>
+  ExtValue genval(const Fortran::evaluate::ArrayConstructor<A> &) {
+    TODO(getLoc(), "genval ArrayConstructor<A>");
+  }
+
+  ExtValue genval(const Fortran::evaluate::ComplexPart &x) {
+    TODO(getLoc(), "genval ComplexPart");
+  }
+
+  ExtValue genval(const Fortran::evaluate::Substring &ss) {
+    TODO(getLoc(), "genval Substring");
+  }
+
+  ExtValue genval(const Fortran::evaluate::Subscript &subs) {
+    TODO(getLoc(), "genval Subscript");
+  }
+
+  ExtValue genval(const Fortran::evaluate::DataRef &dref) {
+    TODO(getLoc(), "genval DataRef");
+  }
+
+  ExtValue genval(const Fortran::evaluate::Component &cmpt) {
+    TODO(getLoc(), "genval Component");
+  }
+
+  ExtValue genval(const Fortran::semantics::Bound &bound) {
+    TODO(getLoc(), "genval Bound");
+  }
+
+  ExtValue genval(const Fortran::evaluate::ArrayRef &aref) {
+    TODO(getLoc(), "genval ArrayRef");
+  }
+
+  ExtValue genval(const Fortran::evaluate::CoarrayRef &coref) {
+    TODO(getLoc(), "genval CoarrayRef");
+  }
+
+  template <typename A>
+  ExtValue genval(const Fortran::evaluate::Designator<A> &des) {
+    TODO(getLoc(), "genval Designator<A>");
+  }
+
+  template <typename A>
+  ExtValue genval(const Fortran::evaluate::FunctionRef<A> &funcRef) {
+    TODO(getLoc(), "genval FunctionRef<A>");
+  }
+
+  ExtValue genval(const Fortran::evaluate::ProcedureRef &procRef) {
+    TODO(getLoc(), "genval ProcedureRef");
+  }
+
+  template <typename A>
+  bool isScalar(const A &x) {
+    return x.Rank() == 0;
+  }
+
+  template <typename A>
+  ExtValue genval(const Fortran::evaluate::Expr<A> &x) {
+    if (isScalar(x))
+      return std::visit([&](const auto &e) { return genval(e); }, x.u);
+    TODO(getLoc(), "genval Expr<A> arrays");
+  }
+
+  template <int KIND>
+  ExtValue genval(const Fortran::evaluate::Expr<Fortran::evaluate::Type<
+                      Fortran::common::TypeCategory::Logical, KIND>> &exp) {
+    return std::visit([&](const auto &e) { return genval(e); }, exp.u);
+  }
+
+private:
+  mlir::Location location;
+  Fortran::lower::AbstractConverter &converter;
+  fir::FirOpBuilder &builder;
+  Fortran::lower::SymMap &symMap;
+};
+} // namespace
+
+fir::ExtendedValue Fortran::lower::createSomeExtendedExpression(
+    mlir::Location loc, Fortran::lower::AbstractConverter &converter,
+    const Fortran::lower::SomeExpr &expr, Fortran::lower::SymMap &symMap) {
+  LLVM_DEBUG(expr.AsFortran(llvm::dbgs() << "expr: ") << '\n');
+  return ScalarExprLowering{loc, converter, symMap}.genval(expr);
+}

flang/lib/Lower/ConvertType.cpp

private:
   //===--------------------------------------------------------------------===//
 
   mlir::MLIRContext *context;
   const Fortran::common::IntrinsicTypeDefaultKinds &defaults;
 };
 
 } // namespace
 
-mlir::Type Fortran::lower::getFIRType(
-    mlir::MLIRContext *context,
-    const Fortran::common::IntrinsicTypeDefaultKinds &defaults,
+template <int KIND>
+int getIntegerBits() {
+  return Fortran::evaluate::Type<Fortran::common::TypeCategory::Integer,
+                                 KIND>::Scalar::bits;
+}
+static mlir::Type genIntegerType(mlir::MLIRContext *context, int kind) {
+  if (Fortran::evaluate::IsValidKindOfIntrinsicType(
+          Fortran::common::TypeCategory::Integer, kind)) {
+    switch (kind) {
+    case 1:
+      return mlir::IntegerType::get(context, getIntegerBits<1>());
+    case 2:
+      return mlir::IntegerType::get(context, getIntegerBits<2>());
+    case 4:
+      return mlir::IntegerType::get(context, getIntegerBits<4>());
+    case 8:
+      return mlir::IntegerType::get(context, getIntegerBits<8>());
+    case 16:
+      return mlir::IntegerType::get(context, getIntegerBits<16>());
+    }
+  }
+  llvm_unreachable("INTEGER kind not translated");
+}
+
+static mlir::Type genFIRType(mlir::MLIRContext *context,
                              Fortran::common::TypeCategory tc, int kind) {
-  return TypeBuilder{context, defaults}.genFIRTy(tc, kind);
+  switch (tc) {
+  case Fortran::common::TypeCategory::Integer:
+    return genIntegerType(context, kind);
+  case Fortran::common::TypeCategory::Real:
+  case Fortran::common::TypeCategory::Complex:
+  case Fortran::common::TypeCategory::Logical:
+  case Fortran::common::TypeCategory::Character:
+  default:
+    break;
+  }
+  llvm_unreachable("unhandled type category");
+}
+
+mlir::Type Fortran::lower::getFIRType(mlir::MLIRContext *context,
+                                      Fortran::common::TypeCategory tc,
+                                      int kind) {
+  return genFIRType(context, tc, kind);
 }
 
 mlir::Type Fortran::lower::getFIRType(
     mlir::MLIRContext *context,
     const Fortran::common::IntrinsicTypeDefaultKinds &defaults,
     Fortran::common::TypeCategory tc) {
   return TypeBuilder{context, defaults}.genFIRTy(tc);
 }