提交 89e9bd6b authored 作者: Ricardo Vieira's avatar Ricardo Vieira 提交者: Ricardo Vieira

Rename file elemwise_codegen to vectorize_codegen

上级 fc21336a
...@@ -18,7 +18,7 @@ from pytensor import config ...@@ -18,7 +18,7 @@ from pytensor import config
from pytensor.graph.basic import Apply from pytensor.graph.basic import Apply
from pytensor.graph.op import Op from pytensor.graph.op import Op
from pytensor.link.numba.dispatch import basic as numba_basic from pytensor.link.numba.dispatch import basic as numba_basic
from pytensor.link.numba.dispatch import elemwise_codegen from pytensor.link.numba.dispatch import vectorize_codegen
from pytensor.link.numba.dispatch.basic import ( from pytensor.link.numba.dispatch.basic import (
create_numba_signature, create_numba_signature,
create_tuple_creator, create_tuple_creator,
...@@ -558,14 +558,14 @@ def _vectorized( ...@@ -558,14 +558,14 @@ def _vectorized(
] ]
in_shapes = [cgutils.unpack_tuple(builder, obj.shape) for obj in inputs] in_shapes = [cgutils.unpack_tuple(builder, obj.shape) for obj in inputs]
iter_shape = elemwise_codegen.compute_itershape( iter_shape = vectorize_codegen.compute_itershape(
ctx, ctx,
builder, builder,
in_shapes, in_shapes,
input_bc_patterns_val, input_bc_patterns_val,
) )
outputs, output_types = elemwise_codegen.make_outputs( outputs, output_types = vectorize_codegen.make_outputs(
ctx, ctx,
builder, builder,
iter_shape, iter_shape,
...@@ -576,7 +576,7 @@ def _vectorized( ...@@ -576,7 +576,7 @@ def _vectorized(
input_types, input_types,
) )
elemwise_codegen.make_loop_call( vectorize_codegen.make_loop_call(
typingctx, typingctx,
ctx, ctx,
builder, builder,
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论