提交 622f1d51 authored 作者: Arnaud Bergeron's avatar Arnaud Bergeron

Fix calls to as_gpuarray_variable.

上级 90d31932
...@@ -753,9 +753,9 @@ def local_gpua_dot22(node, context_name): ...@@ -753,9 +753,9 @@ def local_gpua_dot22(node, context_name):
@register_opt('fast_compile') @register_opt('fast_compile')
@op_lifter([tensor.blas.Dot22Scalar]) @op_lifter([tensor.blas.Dot22Scalar])
def local_gpua_dot22scalar(node, context_name): def local_gpua_dot22scalar(node, context_name):
# x, y, a dot = gpu_dot22(as_gpuarray_variable(node.inputs[0], context_name),
return [node.inputs[2] * gpu_dot22(as_gpuarray_variable(node.inputs[0]), as_gpuarray_variable(node.inputs[1], context_name))
as_gpuarray_variable(node.inputs[1]))] return [node.inputs[2] * dot]
@register_opt('fast_compile') @register_opt('fast_compile')
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论