提交 354c4a9f authored 作者: Arnaud Bergeron's avatar Arnaud Bergeron

Revert back to transferring all inputs by default.

We still have the option to disable that.
上级 e5f7eb3a
...@@ -538,11 +538,9 @@ int, ...) however GPU support varies and some units can't deal with ...@@ -538,11 +538,9 @@ int, ...) however GPU support varies and some units can't deal with
double (float64) or small (less than 32 bits like int16) data types. double (float64) or small (less than 32 bits like int16) data types.
You will get an error at compile time or runtime if this is the case. You will get an error at compile time or runtime if this is the case.
Also, by default float inputs will get transferred to GPU, but int By default all inputs will get transferred to GPU. You can prevent an
will not. You can force the transfer of int inputs by setting the input from getting transferred by setting its tag.target attribute to
tag.target attribute to None or a context name. You can also prevent 'cpu'.
a float value from getting transferred by setting its tag.target
attribute to 'cpu'.
Complex support is untested and most likely completely broken. Complex support is untested and most likely completely broken.
......
...@@ -172,12 +172,7 @@ class InputToGpuOptimizer(Optimizer): ...@@ -172,12 +172,7 @@ class InputToGpuOptimizer(Optimizer):
for cl in input.clients)): for cl in input.clients)):
continue continue
if input.type.dtype.startswith('float'): target = getattr(input.tag, 'target', None)
default = None
else:
default = 'cpu'
target = getattr(input.tag, 'target', default)
if target == 'cpu': if target == 'cpu':
continue continue
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论