提交 d78bce89 authored 作者: Razvan Pascanu's avatar Razvan Pascanu

merge

...@@ -544,7 +544,7 @@ class Scan(Op): ...@@ -544,7 +544,7 @@ class Scan(Op):
part_2 = range(pos[idx] ) part_2 = range(pos[idx] )
reordered = part_1 + part_2 reordered = part_1 + part_2
if len(reordered) > 1: if len(reordered) > 1:
if isinstance( outs[idx][0], cuda.CudaNdarray): if cuda.cuda_available and isinstance( outs[idx][0], cuda.CudaNdarray):
shape = outs[idx][0].shape shape = outs[idx][0].shape
tmp = cuda.cuda_ndarray.cuda_ndarray.CudaNdarray.zeros(shape) tmp = cuda.cuda_ndarray.cuda_ndarray.CudaNdarray.zeros(shape)
pdx = pos[idx] pdx = pos[idx]
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论