提交 a0df82f9 authored 作者: Nicolas Pinto's avatar Nicolas Pinto

FIX: tutorial pycuda <> theano compatilbity

thunk needs lazy attribute
上级 d173f8cb
...@@ -115,6 +115,7 @@ You can use a GPU function compiled with PyCUDA in a Theano op: ...@@ -115,6 +115,7 @@ You can use a GPU function compiled with PyCUDA in a Theano op:
grid = (int(numpy.ceil(inputs[0][0].size / 512.)),1) grid = (int(numpy.ceil(inputs[0][0].size / 512.)),1)
pycuda_fct(inputs[0][0], z[0], numpy.intc(inputs[0][0].size), pycuda_fct(inputs[0][0], z[0], numpy.intc(inputs[0][0].size),
block=(512, 1, 1), grid=grid) block=(512, 1, 1), grid=grid)
thunk.lazy = False
return thunk return thunk
CUDAMat CUDAMat
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论