提交 abf2cb4c authored 作者: Arnaud Bergeron's avatar Arnaud Bergeron

Cover missed case for _dev20

上级 dcfe5881
...@@ -1118,7 +1118,8 @@ def local_gpua_advanced_incsubtensor1(op, context_name, inputs, outputs): ...@@ -1118,7 +1118,8 @@ def local_gpua_advanced_incsubtensor1(op, context_name, inputs, outputs):
set_instead_of_inc = op.set_instead_of_inc set_instead_of_inc = op.set_instead_of_inc
if (x.ndim == 1 and y.ndim == 0 and if (x.ndim == 1 and y.ndim == 0 and
config.deterministic == 'default'): config.deterministic == 'default' and
x.dtype not in ('int8', 'int16')):
x = x.dimshuffle(0, 'x') x = x.dimshuffle(0, 'x')
y = y.dimshuffle('x', 'x') y = y.dimshuffle('x', 'x')
ret = GpuAdvancedIncSubtensor1_dev20( ret = GpuAdvancedIncSubtensor1_dev20(
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论