提交 133b8a48 authored 作者: Arnaud Bergeron's avatar Arnaud Bergeron

Fix cuda opt.

上级 4ed010d8
......@@ -1539,7 +1539,7 @@ def typeConstructor(broadcastable, dtype):
if dtype == 'float32':
return CudaNdarrayType(broadcastable=broadcastable)
else:
return TensorType(broadcastable=broadcastable, dtype=dtype)
return tensor.TensorType(broadcastable=broadcastable, dtype=dtype)
@register_opt('scan')
@local_optimizer([gpu_from_host, scan_op.Scan])
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论