Support checking and codegen of constant vector globals
llvm-svn: 46343
diff --git a/clang/AST/Expr.cpp b/clang/AST/Expr.cpp
index 8aa5eed..becce2a 100644
--- a/clang/AST/Expr.cpp
+++ b/clang/AST/Expr.cpp
@@ -487,7 +487,8 @@
case CompoundLiteralExprClass:
if (Loc) *Loc = getLocStart();
// Allow "(int []){2,4}", since the array will be converted to a pointer.
- return TR->isArrayType();
+ // Allow "(vector type){2,4}" since the elements are all constant.
+ return TR->isArrayType() || TR->isVectorType();
case UnaryOperatorClass: {
const UnaryOperator *Exp = cast<UnaryOperator>(this);
diff --git a/clang/CodeGen/CodeGenModule.cpp b/clang/CodeGen/CodeGenModule.cpp
index bb61dc7..e841ee5 100644
--- a/clang/CodeGen/CodeGenModule.cpp
+++ b/clang/CodeGen/CodeGenModule.cpp
@@ -300,7 +300,8 @@
return 0;
}
- assert((ILE->getType()->isArrayType() || ILE->getType()->isStructureType()) &&
+ assert((ILE->getType()->isArrayType() || ILE->getType()->isStructureType() ||
+ ILE->getType()->isVectorType()) &&
"Bad type for init list!");
CodeGenTypes& Types = CGM.getTypes();
@@ -342,6 +343,9 @@
if (ILE->getType()->isStructureType())
return llvm::ConstantStruct::get(cast<llvm::StructType>(CType), Elts);
+ if (ILE->getType()->isVectorType())
+ return llvm::ConstantVector::get(cast<llvm::VectorType>(CType), Elts);
+
// Make sure we have an array at this point
assert(AType);
@@ -417,6 +421,12 @@
return llvm::ConstantArray::get(Str, false);
}
+ // Generate initializer for the CompoundLiteral
+ case Stmt::CompoundLiteralExprClass: {
+ const CompoundLiteralExpr *CLE = cast<CompoundLiteralExpr>(Expression);
+ return GenerateConstantExpr(CLE->getInitializer(), CGM);
+ }
+
// Elide parenthesis.
case Stmt::ParenExprClass:
return GenerateConstantExpr(cast<ParenExpr>(Expression)->getSubExpr(), CGM);
diff --git a/clang/test/CodeGen/ocu-vector.c b/clang/test/CodeGen/ocu-vector.c
index ee6e737..cffef33 100644
--- a/clang/test/CodeGen/ocu-vector.c
+++ b/clang/test/CodeGen/ocu-vector.c
@@ -3,6 +3,7 @@
typedef __attribute__(( ocu_vector_type(4) )) float float4;
typedef __attribute__(( ocu_vector_type(2) )) float float2;
+float4 foo = (float4){ 1.0, 2.0, 3.0, 4.0 };
float4 test1(float4 V) {
return V.wzyx+V;
diff --git a/clang/test/Sema/vector-init.c b/clang/test/Sema/vector-init.c
new file mode 100644
index 0000000..5436cba
--- /dev/null
+++ b/clang/test/Sema/vector-init.c
@@ -0,0 +1,5 @@
+// RUN: clang %s -verify -fsyntax-only
+
+typedef __attribute__(( ocu_vector_type(4) )) float float4;
+
+float4 foo = (float4){ 1.0, 2.0, 3.0, 4.0 };