Skip to content

T
tic
Commit 7f29f2d4 by Masahiro Sakai Committed by Tianqi Chen
Options

fix wrong type declaration of float64 "log" in intrin_math.py (#1169)

parent 58888b21
Showing with 1 additions and 1 deletions
tutorials/language/intrin_math.py
...@@ -116,7 +116,7 @@ def my_cuda_mylog_rule(op): ...@@ -116,7 +116,7 @@ def my_cuda_mylog_rule(op):
if op.dtype == "float32": if op.dtype == "float32":
return tvm.call_pure_extern("float32", "logf", op.args[0]) return tvm.call_pure_extern("float32", "logf", op.args[0])
elif op.dtype == "float64": 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: else:
return op return op
tvm.register_intrin_rule("cuda", "mylog", my_cuda_mylog_rule, override=True) tvm.register_intrin_rule("cuda", "mylog", my_cuda_mylog_rule, override=True)
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment