aaron/ispc
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

foreach_unique will work now on atomic data types, not pointers yet. enum is not tested. All tests/foreach-unique-*.ispc pass

Browse Source
This commit is contained in:
Evghenii
2014-01-24 08:30:50 +01:00
parent 1cf1dab649
commit 1a07aed6aa
3 changed files with 73 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
stmt.cpp
View File

@@ -2355,8 +2355,6 @@ ForeachUniqueStmt::ForeachUniqueStmt(const char *iterName, Expr *e,
sym = m->symbolTable->LookupVariable(iterName);
expr = e;
stmts = s;
if (g->target->getISA() == Target::NVPTX)
Error(pos, "\"foreach_unique\" is not yetsupported with \"nvptx\" target.");
}
@@ -2442,10 +2440,19 @@ ForeachUniqueStmt::EmitCode(FunctionEmitContext *ctx) const {
// And load the corresponding element value from the temporary
// memory storing the value of the varying expr.
llvm::Value *uniqueValuePtr =
llvm::Value *uniqueValue;
if (g->target->getISA() != Target::NVPTX)
{
llvm::Value *uniqueValuePtr =
ctx->GetElementPtrInst(exprMem, LLVMInt64(0), firstSet, exprPtrType,
"unique_index_ptr");
llvm::Value *uniqueValue = ctx->LoadInst(uniqueValuePtr, "unique_value");
"unique_index_ptr");
uniqueValue = ctx->LoadInst(uniqueValuePtr, "unique_value");
}
else /* in case of PTX target, use __shfl PTX intrinsics via __insert/__extract function */
{
llvm::Value *firstSet32 = ctx->TruncInst(firstSet, LLVMTypes::Int32Type);
uniqueValue = ctx->Extract(exprValue, firstSet32);
}
// If it's a varying pointer type, need to convert from the int
// type we store in the vector to the actual pointer type