Skip to content

Commit

Permalink
Review comments
Browse files Browse the repository at this point in the history
  • Loading branch information
siju-samuel committed Apr 14, 2020
1 parent fa1b859 commit 12c12bc
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion src/target/intrin_rule.cc
Original file line number Diff line number Diff line change
Expand Up @@ -96,7 +96,7 @@ TVM_REGISTER_GLOBAL("tvm.intrin.rule.default.sigmoid")
*rv = one / (one + exp(-call->args[0]));
});

TVM_REGISTER_GLOBAL("tvm.intrin.rule.default.nan")
TVM_REGISTER_GLOBAL("tvm.intrin.rule.default.isnan")
.set_body([](const TVMArgs& args, TVMRetValue* rv){
PrimExpr e = args[0];
const CallNode* call = e.as<CallNode>();
Expand Down

0 comments on commit 12c12bc

Please sign in to comment.