提交 ca19b615 authored 作者: Harm de Vries's avatar Harm de Vries

update prepare_node blas

上级 5212d681
...@@ -1953,7 +1953,7 @@ class GpuConv(GpuOp): ...@@ -1953,7 +1953,7 @@ class GpuConv(GpuOp):
images[2] * images[3] * 2) images[2] * images[3] * 2)
return flops return flops
def prepare_node(self, node): def prepare_node(self, node, storage_map, compute_map):
if node.op.max_threads_dim0 is None: if node.op.max_threads_dim0 is None:
cuda = theano.sandbox.cuda cuda = theano.sandbox.cuda
device_id = cuda.use.device_number device_id = cuda.use.device_number
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论