Skip to content

Commit

Permalink
fix wrong type declaration of float64 "log" in intrin_math.py (apache…
Browse files Browse the repository at this point in the history
  • Loading branch information
msakai authored and tqchen committed May 21, 2018
1 parent 58888b2 commit 7f29f2d
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion tutorials/language/intrin_math.py
Original file line number Diff line number Diff line change
Expand Up @@ -116,7 +116,7 @@ def my_cuda_mylog_rule(op):
if op.dtype == "float32":
return tvm.call_pure_extern("float32", "logf", op.args[0])
elif op.dtype == "float64":
return tvm.call_pure_extern("float32", "log", op.args[0])
return tvm.call_pure_extern("float64", "log", op.args[0])
else:
return op
tvm.register_intrin_rule("cuda", "mylog", my_cuda_mylog_rule, override=True)
Expand Down

0 comments on commit 7f29f2d

Please sign in to comment.