提交 bcbbdf5e authored 作者: Shawn Tan's avatar Shawn Tan

Added convert_f16=True flag for elemwise2

上级 f71035bf
......@@ -690,7 +690,8 @@ class GpuAdvancedIncSubtensor(HideC, tensor.AdvancedIncSubtensor):
x_flat_sub = x_flat[i]
tmp = pygpu.elemwise.elemwise2(
x_flat_sub, '+', y_flat, x_flat_sub,
broadcast=True
broadcast=True,
convert_f16=True
)
x_flat[i].__setitem__(index, tmp)
else:
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论