[CIR][Lowering] Support lowering of nested string ConstantArrays.

llvm · Aug 24, 2023 · e501bbc · e501bbc
1 parent 2fae9b7
commit e501bbc
Show file tree

Hide file tree

Showing 4 changed files with 29 additions and 5 deletions.
diff --git a/clang/lib/CIR/CodeGen/CIRGenBuilder.h b/clang/lib/CIR/CodeGen/CIRGenBuilder.h
@@ -244,6 +244,8 @@ class CIRGenBuilderTy : public mlir::OpBuilder {
     }
 
     if (const auto arrayVal = attr.dyn_cast<mlir::cir::ConstArrayAttr>()) {
+      if (arrayVal.getElts().isa<mlir::StringAttr>())
+        return false;
       for (const auto elt : arrayVal.getElts().cast<mlir::ArrayAttr>()) {
         if (!isNullValue(elt))
           return false;

diff --git a/clang/lib/CIR/Lowering/DirectToLLVM/LowerToLLVM.cpp b/clang/lib/CIR/Lowering/DirectToLLVM/LowerToLLVM.cpp
@@ -141,12 +141,26 @@ mlir::Value lowerCirAttrAsValue(mlir::cir::ConstArrayAttr constArr,
                                 mlir::TypeConverter *converter) {
   auto llvmTy = converter->convertType(constArr.getType());
   mlir::Value result = rewriter.create<mlir::LLVM::UndefOp>(loc, llvmTy);
-  auto arrayAttr = constArr.getElts().cast<mlir::ArrayAttr>();
 
   // Iteratively lower each constant element of the array.
-  for (auto [idx, elt] : llvm::enumerate(arrayAttr)) {
-    mlir::Value init = lowerCirAttrAsValue(elt, loc, rewriter, converter);
-    result = rewriter.create<mlir::LLVM::InsertValueOp>(loc, result, init, idx);
+  if (auto arrayAttr = constArr.getElts().dyn_cast<mlir::ArrayAttr>()) {
+    for (auto [idx, elt] : llvm::enumerate(arrayAttr)) {
+      mlir::Value init = lowerCirAttrAsValue(elt, loc, rewriter, converter);
+      result =
+          rewriter.create<mlir::LLVM::InsertValueOp>(loc, result, init, idx);
+    }
+  } else if (auto strAttr = constArr.getElts().dyn_cast<mlir::StringAttr>()) {
+    auto arrayTy = strAttr.getType().dyn_cast<mlir::cir::ArrayType>();
+    assert(arrayTy && "String attribute must have an array type");
+    auto eltTy = arrayTy.getEltType();
+    for (auto [idx, elt] : llvm::enumerate(strAttr)) {
+      auto init = rewriter.create<mlir::LLVM::ConstantOp>(
+          loc, converter->convertType(eltTy), elt);
+      result =
+          rewriter.create<mlir::LLVM::InsertValueOp>(loc, result, init, idx);
+    }
+  } else {
+    llvm_unreachable("unexpected ConstArrayAttr elements");
   }
 
   return result;

diff --git a/clang/test/CIR/CodeGen/globals.c b/clang/test/CIR/CodeGen/globals.c
@@ -33,7 +33,12 @@ struct {
   int x;
   int y[2][2];
 } nestedTwoDim = {1, {{2, 3}, {4, 5}}};
-// CHECK: cir.global external @nestedTwoDim = #cir.const_struct<{#cir.int<1> : !s32i, #cir.const_array<[#cir.const_array<[#cir.int<2> : !s32i, #cir.int<3> : !s32i]> : !cir.array<!s32i x 2>, #cir.const_array<[#cir.int<4> : !s32i, #cir.int<5> : !s32i]> : !cir.array<!s32i x 2>]> : !cir.array<!cir.array<!s32i x 2> x 2>}> : !ty_22struct2Eanon22
+// CHECK: cir.global external @nestedTwoDim = #cir.const_struct<{#cir.int<1> : !s32i, #cir.const_array<[#cir.const_array<[#cir.int<2> : !s32i, #cir.int<3> : !s32i]> : !cir.array<!s32i x 2>, #cir.const_array<[#cir.int<4> : !s32i, #cir.int<5> : !s32i]> : !cir.array<!s32i x 2>]> : !cir.array<!cir.array<!s32i x 2> x 2>}>
+
+struct {
+  char x[3];
+} nestedString = {"1"};
+// CHECK: cir.global external @nestedString = #cir.const_struct<{#cir.const_array<"1\00\00" : !cir.array<!s8i x 3>> : !cir.array<!s8i x 3>}>
 
 // TODO: test tentatives with internal linkage.
 

diff --git a/clang/test/CIR/Lowering/globals.cir b/clang/test/CIR/Lowering/globals.cir
@@ -12,6 +12,7 @@
 !u8i = !cir.int<u, 8>
 !ty_22struct2EA22 = !cir.struct<"struct.A", !s32i, !cir.array<!cir.array<!s32i x 2> x 2>, #cir.recdecl.ast>
 !ty_22struct2EBar22 = !cir.struct<"struct.Bar", !s32i, !s8i, #cir.recdecl.ast>
+!ty_22struct2Eanon22 = !cir.struct<"struct.anon", !cir.array<!s8i x 3>, #cir.recdecl.ast>
 
 module {
   cir.global external @a = #cir.int<3> : !s32i
@@ -91,6 +92,8 @@ module {
   // LLVM: @twoDim = global [2 x [2 x i32{{\]\] \[\[}}2 x i32] [i32 1, i32 2], [2 x i32] [i32 3, i32 4{{\]\]}}
   cir.global external @nestedTwoDim = #cir.const_struct<{#cir.int<1> : !s32i, #cir.const_array<[#cir.const_array<[#cir.int<2> : !s32i, #cir.int<3> : !s32i]> : !cir.array<!s32i x 2>, #cir.const_array<[#cir.int<4> : !s32i, #cir.int<5> : !s32i]> : !cir.array<!s32i x 2>]> : !cir.array<!cir.array<!s32i x 2> x 2>}> : !ty_22struct2EA22
   // LLVM: @nestedTwoDim = global %struct.A { i32 1, [2 x [2 x i32{{\]\] \[\[}}2 x i32] [i32 2, i32 3], [2 x i32] [i32 4, i32 5{{\]\]}} }
+  cir.global external @nestedString = #cir.const_struct<{#cir.const_array<"1\00\00" : !cir.array<!s8i x 3>> : !cir.array<!s8i x 3>}> : !ty_22struct2Eanon22
+  // LLVM: @nestedString = global %struct.anon { [3 x i8] c"1\00\00" }
   cir.func @_Z11get_globalsv() {
     %0 = cir.alloca !cir.ptr<!s8i>, cir.ptr <!cir.ptr<!s8i>>, ["s", init] {alignment = 8 : i64}
     %1 = cir.alloca !cir.ptr<!u32i>, cir.ptr <!cir.ptr<!u32i>>, ["u", init] {alignment = 8 : i64}