提交 b887bd78 authored 作者: Ricardo Vieira's avatar Ricardo Vieira 提交者: Ricardo Vieira

Align numba reciprocal with C backend

上级 4e3b5bed
...@@ -258,11 +258,10 @@ def numba_funcify_Second(op, node, **kwargs): ...@@ -258,11 +258,10 @@ def numba_funcify_Second(op, node, **kwargs):
def numba_funcify_Reciprocal(op, node, **kwargs): def numba_funcify_Reciprocal(op, node, **kwargs):
@numba_basic.numba_njit @numba_basic.numba_njit
def reciprocal(x): def reciprocal(x):
# TODO FIXME: This isn't really the behavior or `numpy.reciprocal` when # This is how the C-backend implementation works
# `x` is an `int` return np.divide(np.float32(1.0), x)
return 1 / x
return reciprocal, scalar_op_cache_key(op) return reciprocal, scalar_op_cache_key(op, cache_version=1)
@register_funcify_and_cache_key(Sigmoid) @register_funcify_and_cache_key(Sigmoid)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论