Handle allocation when allocating struct with new (#227)

Author: chelini

tools/cgeist/Lib/clang-mlir.cc

@@ -1199,6 +1199,9 @@ ValueCategory MLIRScanner::VisitCXXNewExpr(clang::CXXNewExpr *expr) {
     auto shape = std::vector<int64_t>(mt.getShape());
     mlir::Value args[1] = {count};
     arrayCons = alloc = builder.create<mlir::memref::AllocOp>(loc, mt, args);
+    if (expr->hasInitializer() && isa<InitListExpr>(expr->getInitializer()))
+      (void)InitializeValueByInitListExpr(alloc, expr->getInitializer());
+
   } else {
     auto i64 = mlir::IntegerType::get(count.getContext(), 64);
     auto typeSize = getTypeSize(expr->getAllocatedType());

tools/cgeist/Test/Verification/new.cpp

@@ -0,0 +1,20 @@
+// RUN: cgeist %s %stdinclude --function=* -S | FileCheck %s
+
+#include <stdio.h>
+
+struct A {
+  float x, y;
+};
+
+void f(A *a) { printf("a.x = %f, a.y = %f\n", a->x, a->y); }
+
+int main(int argc, char const *argv[]) {
+  // CHECK-DAG: %[[two:.*]] = arith.constant 2.000000e+00 : f32
+  // CHECK-DAG: %[[one:.*]] = arith.constant 1.000000e+00 : f32
+  // CHECK: %[[alloc:.*]] = memref.alloc() : memref<1x2xf32>
+  // CHECK: affine.store %[[one]], %[[alloc]][0, 0] : memref<1x2xf32>
+  // CHECK: affine.store %[[two]], %[[alloc]][0, 1] : memref<1x2xf32>
+  auto *a = new A{1.0f, 2.0f};
+  f(a);
+  return 0;
+}