提交 ce684a87 authored 作者: Caglar's avatar Caglar

fix casting in advanced subtensor.

上级 b524dcfb
...@@ -2567,7 +2567,8 @@ class GpuAdvancedSubtensor1(tensor.AdvancedSubtensor1, GpuOp): ...@@ -2567,7 +2567,8 @@ class GpuAdvancedSubtensor1(tensor.AdvancedSubtensor1, GpuOp):
# c code suppose it is int64 # c code suppose it is int64
if x.ndim in [2, 3] and ilist_.dtype in [ if x.ndim in [2, 3] and ilist_.dtype in [
'int8', 'int16', 'int32', 'uint8', 'uint16', 'uint32']: 'int8', 'int16', 'int32', 'uint8', 'uint16', 'uint32']:
ilist_ = ilist_.cast('int64') ilist_ = tensor.cast(ilist_, 'int64')
bcast = (ilist_.broadcastable[0],) + x_.broadcastable[1:] bcast = (ilist_.broadcastable[0],) + x_.broadcastable[1:]
return Apply(self, [x_, ilist_], return Apply(self, [x_, ilist_],
[CudaNdarrayType(dtype=x.dtype, [CudaNdarrayType(dtype=x.dtype,
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论