提交 e0ab0b46 authored 作者: Brandon T. Willard's avatar Brandon T. Willard 提交者: Brandon T. Willard

Remove Python-only BroadcastTo Numba conversion

上级 4e617870
...@@ -48,7 +48,6 @@ from aesara.tensor.basic import ( ...@@ -48,7 +48,6 @@ from aesara.tensor.basic import (
from aesara.tensor.elemwise import CAReduce, DimShuffle, Elemwise from aesara.tensor.elemwise import CAReduce, DimShuffle, Elemwise
from aesara.tensor.extra_ops import ( from aesara.tensor.extra_ops import (
Bartlett, Bartlett,
BroadcastTo,
CumOp, CumOp,
DiffOp, DiffOp,
FillDiagonal, FillDiagonal,
...@@ -1316,21 +1315,3 @@ def numba_funcify_Searchsorted(op, node, **kwargs): ...@@ -1316,21 +1315,3 @@ def numba_funcify_Searchsorted(op, node, **kwargs):
return np.searchsorted(a, v, side) return np.searchsorted(a, v, side)
return searchsorted return searchsorted
@numba_funcify.register(BroadcastTo)
def numba_funcify_BroadcastTo(op, node, **kwargs):
warnings.warn(
("Numba will use object mode to allow the " "use of `numpy.broadcast_to`."),
UserWarning,
)
ret_sig = get_numba_type(node.outputs[0].type)
@numba.njit
def broadcastto(x, *shape):
with numba.objmode(ret=ret_sig):
ret = np.broadcast_to(x, shape)
return ret
return broadcastto
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论