提交 ad16bb64 authored 作者: Frederic Bastien's avatar Frederic Bastien

Don't try to insert gpu op that are not float32. This remove too much false warning.

上级 1b083473
...@@ -553,7 +553,7 @@ def local_gpu_advanced_incsubtensor1(node): ...@@ -553,7 +553,7 @@ def local_gpu_advanced_incsubtensor1(node):
gpu_from_host(y), *coords)] gpu_from_host(y), *coords)]
# Should not execute for GpuAdvancedIncSubtensor1 # Should not execute for GpuAdvancedIncSubtensor1
if node.op.__class__ is tensor.AdvancedSubtensor1: if node.op.__class__ is tensor.AdvancedSubtensor1 and node.inputs[0].dtype=="float32":
x, y = node.inputs[0:2] x, y = node.inputs[0:2]
coords = node.inputs[2:] coords = node.inputs[2:]
go_gpu = False go_gpu = False
...@@ -585,7 +585,7 @@ def local_gpu_incsubtensor(node): ...@@ -585,7 +585,7 @@ def local_gpu_incsubtensor(node):
gpu_from_host(x), gpu_from_host(x),
gpu_from_host(y), gpu_from_host(y),
*coords)] *coords)]
if type(node.op) == tensor.IncSubtensor: if type(node.op) == tensor.IncSubtensor and node.inputs[0].dtype=="float32":
x, y = node.inputs[0:2] x, y = node.inputs[0:2]
assert isinstance(x.type, tensor.TensorType) assert isinstance(x.type, tensor.TensorType)
assert isinstance(y.type, tensor.TensorType) assert isinstance(y.type, tensor.TensorType)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论