提交 de44d06d authored 作者: Frederic's avatar Frederic

Don't try to push dot on the gpu when it do some upcast of the input.

Other change triggered this hidded crash.
上级 81957b6d
......@@ -243,6 +243,11 @@ def local_gpu_dot_to_dot22(node):
A more suitable solution would be to use the right cublas call
"""
# In case the got do input upcast, we much check that we can
# make it run on the gpu.
if node.outputs[0].dtype != 'float32':
return False
if node.op == gpu_from_host:
host_input = node.inputs[0]
if host_input.owner and host_input.owner.op == tensor.basic.dot:
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论