gaspersic/triton
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

[LANG] Added sqrtf support

Browse Source
This commit is contained in:
Philippe Tillet
2020-11-17 01:26:53 -05:00
parent 073fddffc1
commit 7d095ec686
6 changed files with 6 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
lib/lang/code_gen.cc
View File

@@ -204,6 +204,7 @@ void Generator::VisitUnaryOp(UnaryOp* unary) {
case Token::CAST: return set_ret(GenSemCastOp(arg, GenIRType(unary->Type(), *ctx_)));
case Token::EXP: return set_ret(bld_->create_exp(arg)); //FIXME cast
case Token::LOG: return set_ret(bld_->create_log(arg));
case Token::SQRTF: return set_ret(bld_->create_sqrt(arg));
case Token::REDUCE: {
int ax, tag;
UnaryOp::decodeRed(unary->info_, ax, tag);
@@ -287,11 +288,6 @@ void Generator::VisitFuncCall(FuncCall* funcCall) {
ir::value* msk = ret_;
return set_ret(bld_->create_atomic_add(ptr, val, msk));
}
if(name == "sqrtf"){
VisitExpr(funcCall->Args()->at(0));
ir::value* ret = ret_;
return set_ret(bld_->create_sqrt(ret));
}
if(name == "calloc"){
VisitExpr(funcCall->Args()->at(0));
ir::value* ret = ret_;