Commit be870567 by liyinqiao

Bug fixed for Gather functions.

parent 779351ac
......@@ -139,7 +139,7 @@ void _CudaGather(const XTensor * s, XTensor * t, XTensor * srcIndex)
}
else {
int * sIndexData = new int[sizeof(int) * indexSize];
XMemCopy(sIndexData, -1, srcIndex, srcIndex->devID, sizeof(int) * indexSize);
XMemCopy(sIndexData, -1, srcIndex->data, srcIndex->devID, sizeof(int) * indexSize);
for (int i = 0; i < indexSize; i++) {
int srcIndexValue = sIndexData[i] * stride;
CheckNTErrors(srcIndexValue < s->unitNum, "Wrong index!");
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论