fix wrong type declaration of float64 "log" in intrin_math.py (#1169)
This commit is contained in:
Родитель
58888b2170
Коммит
7f29f2d46d
|
@ -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)
|
||||
|
|
Загрузка…
Ссылка в новой задаче