[CIR] Implement Atomic init for ComplexType (#1873)

Implement Atomic init for ComplexType

Author: AmrDeveloper

clang/lib/CIR/CodeGen/CIRGenAtomic.cpp

@@ -290,7 +290,9 @@ bool AtomicInfo::requiresMemSetZero(mlir::Type ty) const {
   case cir::TEK_Scalar:
     return !isFullSizeType(CGF.CGM, ty, AtomicSizeInBits);
   case cir::TEK_Complex:
-    llvm_unreachable("NYI");
+    return !isFullSizeType(CGF.CGM,
+                           mlir::cast<cir::ComplexType>(ty).getElementType(),
+                           AtomicSizeInBits / 2);
 
   // Padding in structs has an undefined bit pattern.  User beware.
   case cir::TEK_Aggregate:
@@ -1499,7 +1501,8 @@ void CIRGenFunction::emitAtomicInit(Expr *init, LValue dest) {
   }
 
   case cir::TEK_Complex: {
-    llvm_unreachable("NYI");
+    mlir::Value value = emitComplexExpr(init);
+    atomics.emitCopyIntoMemory(RValue::get(value));
     return;
   }
 

clang/test/CIR/CodeGen/complex.cpp

@@ -210,3 +210,21 @@ void complex_cxx_default_init_expr() {
 // OGCG: %[[C_IMAG_PTR:.*]] = getelementptr inbounds nuw { float, float }, ptr %[[C_ADDR]], i32 0, i32 1
 // OGCG: store float 0.000000e+00, ptr %[[C_REAL_PTR]], align 4
 // OGCG: store float 0.000000e+00, ptr %[[C_IMAG_PTR]], align 4
+
+void complex_init_atomic() {
+  _Atomic(float _Complex) a;
+  __c11_atomic_init(&a, {1.0f, 2.0f});
+}
+
+// CIR: %[[A_ADDR:.*]] = cir.alloca !cir.complex<!cir.float>, !cir.ptr<!cir.complex<!cir.float>>, ["a"]
+// CIR: %[[CONST_COMPLEX:.*]] = cir.const #cir.complex<#cir.fp<1.000000e+00> : !cir.float, #cir.fp<2.000000e+00> : !cir.float> : !cir.complex<!cir.float>
+// CIR: cir.store{{.*}} %[[CONST_COMPLEX]], %[[A_ADDR]] : !cir.complex<!cir.float>, !cir.ptr<!cir.complex<!cir.float>>
+
+// LLVM: %[[A_ADDR:.*]] = alloca { float, float }, i64 1, align 8
+// LLVM: store { float, float } { float 1.000000e+00, float 2.000000e+00 }, ptr %[[A_ADDR]], align 8
+
+// OGCG: %[[A_ADDR:.*]] = alloca { float, float }, align 8
+// OGCG: %[[A_REAL_PTR:.*]] = getelementptr inbounds nuw { float, float }, ptr %[[A_ADDR]], i32 0, i32 0
+// OGCG: %[[A_IMAG_PTR:.*]] = getelementptr inbounds nuw { float, float }, ptr %[[A_ADDR]], i32 0, i32 1
+// OGCG: store float 1.000000e+00, ptr %[[A_REAL_PTR]], align 8
+// OGCG: store float 2.000000e+00, ptr %[[A_IMAG_PTR]], align 4