提交 52d82435 authored 作者: Arnaud Bergeron's avatar Arnaud Bergeron

Don't use the dev20 version for small ints

上级 49c8b94e
...@@ -1126,7 +1126,8 @@ def local_gpua_advanced_incsubtensor1(op, context_name, inputs, outputs): ...@@ -1126,7 +1126,8 @@ def local_gpua_advanced_incsubtensor1(op, context_name, inputs, outputs):
ret = GpuDimShuffle(ret.type.broadcastable, [0])(ret) ret = GpuDimShuffle(ret.type.broadcastable, [0])(ret)
return ret return ret
elif (x.ndim != 2 or y.ndim != 2 or elif (x.ndim != 2 or y.ndim != 2 or
config.deterministic == 'more'): config.deterministic == 'more' or
x.dtype in ('int8', 'int16')):
return GpuAdvancedIncSubtensor1( return GpuAdvancedIncSubtensor1(
set_instead_of_inc=set_instead_of_inc) set_instead_of_inc=set_instead_of_inc)
else: else:
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论