more check

source/tensor/core/movement/CopyIndexed.cpp

@@ -70,7 +70,7 @@ void _CopyIndexed(const XTensor * s, XTensor * t, int dim,
     for (int i = dimRDI; i < t->order; i++)
         blockNumTgt *= t->dimSizeRDI[i];
 
-    CheckNTErrors((blockSizeSrc == blockSizeTgt), "Unmatched tensors!");
+    CheckNTErrors(blockSizeSrc == blockSizeTgt, "Unmatched tensors!");
     indexOffsetNum = blockNumSrc / s->dimSizeRDI[dimRDI];
 
     int realIndexSize = indexOffsetNum * indexSize * copyNum;
@@ -87,13 +87,15 @@ void _CopyIndexed(const XTensor * s, XTensor * t, int dim,
             for (int k = 0; k < copyNum; k++) {
                 rsi[k] = baseSrc + srcIndex[j] + k;
                 rti[k] = baseTgt + tgtIndex[j] + k;
+                CheckNTErrors(rsi[k] < s->unitNum, "Wrong index!");
+                CheckNTErrors(rti[k] < t->unitNum, "Wrong index!");
             }
         }
     }
 
     for (int i = 0; i < indexSize; i++) {
-        CheckNTErrors((srcIndex[i] < blockNumSrc), "Index is out of scope!");
-        CheckNTErrors((tgtIndex[i] < blockNumTgt), "Index is out of scope!");
+        CheckNTErrors(srcIndex[i] < blockNumSrc, "Index is out of scope!");
+        CheckNTErrors(tgtIndex[i] < blockNumTgt, "Index is out of scope!");
     }
 
     _CopyBlocks(s->data, blockSizeSrc * s->unitSize, realSrcIndex, realIndexSize, t->data, realTgtIndex, s->mem, s->devID);