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