Added pragma shufflevector.

This commit is contained in:
Jernej Krempuš
2012-08-05 18:51:49 +02:00
parent 3d108ee684
commit 100907ffa2
4 changed files with 46 additions and 1 deletions

View File

@@ -272,6 +272,7 @@ Msgtable msgtable[] =
{ "no_typeinfo" },
{ "no_moduleinfo" },
{ "Alloca", "alloca" },
{ "Shufflevector", "shufflevector" },
{ "vastart", "va_start" },
{ "vacopy", "va_copy" },
{ "vaend", "va_end" },

View File

@@ -106,6 +106,17 @@ Pragma DtoGetPragma(Scope *sc, PragmaDeclaration *decl, std::string &arg1str)
return LLVMalloca;
}
// pragma(shufflevector) { funcdecl(s) }
else if (ident == Id::Shufflevector)
{
if (args && args->dim > 0)
{
error("takes no parameters");
fatal();
}
return LLVMshufflevector;
}
// pragma(va_start) { templdecl(s) }
else if (ident == Id::vastart)
{
@@ -356,6 +367,18 @@ void DtoCheckPragma(PragmaDeclaration *decl, Dsymbol *s,
}
break;
case LLVMshufflevector:
if (FuncDeclaration* fd = s->isFuncDeclaration())
{
fd->llvmInternal = llvm_internal;
}
else
{
error("the '%s' pragma must only be used on function declarations.", ident->toChars());
fatal();
}
break;
case LLVMinline_asm:
if (TemplateDeclaration* td = s->isTemplateDeclaration())
{

View File

@@ -14,6 +14,7 @@ enum Pragma
LLVMno_typeinfo,
LLVMno_moduleinfo,
LLVMalloca,
LLVMshufflevector,
LLVMva_start,
LLVMva_copy,
LLVMva_end,

View File

@@ -947,8 +947,28 @@ DValue* CallExp::toElem(IRState* p)
if (expv->getType()->toBasetype()->ty != Tint32)
expv = DtoCast(loc, expv, Type::tint32);
return new DImValue(type, p->ir->CreateAlloca(LLType::getInt8Ty(gIR->context()), expv->getRVal(), ".alloca"));
}
// shufflevector
else if (fndecl->llvmInternal == LLVMshufflevector) {
llvm::SmallVector<llvm::Constant*, 32> mask;
for(int i = 2, n = arguments->dim; i < n; i++){
Expression* exp = static_cast<Expression*>(arguments->data[i]);
if(exp->op != TOKint64){
error("Function %s was declared with pragma shufflevector. Because of that all of its arguments except the first two must be integer literals.", f->toChars());
fatal();
}
IntegerExp* iexp = static_cast<IntegerExp*>(arguments->data[i]);
mask.push_back(iexp->toConstElem(p));
}
LLValue* maskVal = llvm::ConstantVector::get(mask);
Expression* exp1 = static_cast<Expression*>(arguments->data[0]);
Expression* exp2 = static_cast<Expression*>(arguments->data[1]);
LLValue* v1 = exp1->toElem(p)->getRVal();
LLValue* v2 = exp2->toElem(p)->getRVal();
return new DImValue(type, p->ir->CreateShuffleVector(v1, v2, maskVal));
}
// fence instruction
} else if (fndecl->llvmInternal == LLVMfence) {
else if (fndecl->llvmInternal == LLVMfence) {
if (arguments->dim != 1) {
error("fence instruction expects 1 arguments");
return NULL;