提交 eb742ddb authored 作者: Iban Harlouchet's avatar Iban Harlouchet 提交者: Iban Harlouchet

commenting out and unused var in theano/misc/pycuda_example.py

上级 28765bd6
...@@ -360,9 +360,9 @@ class PycudaElemwiseSourceModuleMakeThunkOp(Op): ...@@ -360,9 +360,9 @@ class PycudaElemwiseSourceModuleMakeThunkOp(Op):
else: else:
grid = (1, 1) grid = (1, 1)
block = (inputs[0][0].shape[0], inputs[0][0].shape[1], 1) block = (inputs[0][0].shape[0], inputs[0][0].shape[1], 1)
out = pycuda_fct(inputs[0][0], inputs[1][0], z[0], pycuda_fct(inputs[0][0], inputs[1][0], z[0],
numpy.intc(inputs[1][0].size), block=block, numpy.intc(inputs[1][0].size), block=block,
grid=grid) grid=grid)
thunk.inputs = inputs thunk.inputs = inputs
thunk.outputs = outputs thunk.outputs = outputs
thunk.lazy = False thunk.lazy = False
...@@ -390,12 +390,12 @@ def local_pycuda_gpu_elemwise(node): ...@@ -390,12 +390,12 @@ def local_pycuda_gpu_elemwise(node):
pycuda_optimizer.register("local_pycuda_gpu_elemwise", pycuda_optimizer.register("local_pycuda_gpu_elemwise",
local_pycuda_gpu_elemwise) local_pycuda_gpu_elemwise)
"""
@local_optimizer([GpuElemwise]) @local_optimizer([GpuElemwise])
def local_pycuda_gpu_elemwise_kernel(node): def local_pycuda_gpu_elemwise_kernel(node):
""" ""
GpuElemwise -> PycudaElemwiseKernelOp GpuElemwise -> PycudaElemwiseKernelOp
""" ""
if isinstance(node.op, GpuElemwise): if isinstance(node.op, GpuElemwise):
if not any([any(i.type.broadcastable) for i in node.inputs]): if not any([any(i.type.broadcastable) for i in node.inputs]):
new_op = PycudaElemwiseKernelOp(node.op.scalar_op, new_op = PycudaElemwiseKernelOp(node.op.scalar_op,
...@@ -405,3 +405,4 @@ def local_pycuda_gpu_elemwise_kernel(node): ...@@ -405,3 +405,4 @@ def local_pycuda_gpu_elemwise_kernel(node):
pycuda_optimizer.register("local_pycuda_gpu_elemwise_kernel", pycuda_optimizer.register("local_pycuda_gpu_elemwise_kernel",
local_pycuda_gpu_elemwise_kernel, 1.5) local_pycuda_gpu_elemwise_kernel, 1.5)
"""
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论