提交 a50a7986 authored 作者: Arnaud Bergeron's avatar Arnaud Bergeron

Last fix for the import problem.

上级 2d1e254e
......@@ -109,6 +109,7 @@ class CuIFFTOp(ScikitsCudaOp):
broadcastable=[False] * (inp.type.ndim - 1))
def make_thunk(self, node, storage_map, _, _2):
from theano.misc.pycuda_utils import to_gpuarray
inputs = [storage_map[v] for v in node.inputs]
outputs = [storage_map[v] for v in node.outputs]
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论