提交 2bce2e60 authored 作者: abergeron's avatar abergeron

Merge pull request #4439 from nouiz/compat

Compatibility between both GPU back-end and doc.
...@@ -42,6 +42,8 @@ class CudaNdarrayType(Type): ...@@ -42,6 +42,8 @@ class CudaNdarrayType(Type):
dtype = 'float32' dtype = 'float32'
context_name = 'gpu' # For similar interface with the new back-end.
Variable = None Variable = None
""" """
This will be set to the Variable type corresponding to this class. This will be set to the Variable type corresponding to this class.
......
...@@ -224,6 +224,7 @@ class GpuElemwise(HideC, Elemwise): ...@@ -224,6 +224,7 @@ class GpuElemwise(HideC, Elemwise):
// +1 is so that MSVC is happy when nd == 0 // +1 is so that MSVC is happy when nd == 0
size_t dims[%(nd)s+1] = {%(initial_dims)s}; size_t dims[%(nd)s+1] = {%(initial_dims)s};
void *rargs[%(nargs)s] = {0}; void *rargs[%(nargs)s] = {0};
int err;
""" % locals() """ % locals()
for idx, iname in enumerate(inputs): for idx, iname in enumerate(inputs):
if iname in emitted_inames: if iname in emitted_inames:
......
...@@ -255,7 +255,7 @@ def local_cut_gpu_transfers(node): ...@@ -255,7 +255,7 @@ def local_cut_gpu_transfers(node):
return [node.op(n2.inputs[0])] return [node.op(n2.inputs[0])]
gpu_cut_copies.register('cut_gpua_host_transfers', local_cut_gpu_transfers, gpu_cut_copies.register('cut_gpua_host_transfers', local_cut_gpu_transfers,
'fast_compile', 'fast_run', 'inplace', 'gpuarray') 'fast_compile', 'fast_run', 'gpuarray')
gpu_cut_copies.register('cut_gpua_constant_transfers', gpu_cut_copies.register('cut_gpua_constant_transfers',
tensor.opt.constant_folding, tensor.opt.constant_folding,
'fast_compile', 'fast_run', 'gpuarray') 'fast_compile', 'fast_run', 'gpuarray')
......
...@@ -34,7 +34,7 @@ class TensorType(Type): ...@@ -34,7 +34,7 @@ class TensorType(Type):
Optional name for this type. Optional name for this type.
""" """
context_name = 'cpu'
filter_checks_isfinite = False filter_checks_isfinite = False
""" """
When this is True, strict filtering rejects data containing NaN or When this is True, strict filtering rejects data containing NaN or
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论