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

This commit is contained in:
Masahiro Sakai 2018-05-21 12:59:40 +09:00 коммит произвёл Tianqi Chen
Родитель 58888b2170
Коммит 7f29f2d46d
1 изменённых файлов: 1 добавлений и 1 удалений

Просмотреть файл

@ -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)