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

GpuAdvancedIncSubtensor1_dev20 doesn't deal correctly with mixed type inputs.

上级 87a5e2bd
...@@ -809,7 +809,7 @@ class GpuAdvancedIncSubtensor1_dev20(GpuKernelBase, HideC, ...@@ -809,7 +809,7 @@ class GpuAdvancedIncSubtensor1_dev20(GpuKernelBase, HideC,
""" """
ctx_name = infer_context_name(x, y, ilist) ctx_name = infer_context_name(x, y, ilist)
x_ = as_gpuarray_variable(x, ctx_name) x_ = as_gpuarray_variable(x, ctx_name)
y_ = as_gpuarray_variable(y, ctx_name) y_ = as_gpuarray_variable(y.astype(x.dtype), ctx_name)
ilist_ = as_gpuarray_variable(ilist, ctx_name) ilist_ = as_gpuarray_variable(ilist, ctx_name)
assert x_.type.ndim >= y_.type.ndim assert x_.type.ndim >= y_.type.ndim
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论