[Scala/Java] Remove unnecessary data slicing

apache · Feb 6, 2020 · 8158851 · 8158851
1 parent 0392514
commit 8158851
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/scala-package/core/src/main/scala/org/apache/mxnet/module/DataParallelExecutorGroup.scala b/scala-package/core/src/main/scala/org/apache/mxnet/module/DataParallelExecutorGroup.scala
@@ -33,7 +33,7 @@ private object DataParallelExecutorGroup {
                                majorAxis: Seq[Int]): Unit = {
     for (((dSrc, dTargets), axis) <- data zip targets zip majorAxis) {
       for (((sliceIdxStart, sliceIdxStop), dDst) <- dTargets) {
-        if (axis >= 0) {
+        if (axis >= 0 && (sliceIdxStart > 0 || sliceIdxStop < dSrc.shape(axis))) {
           // copy slice
           val shape = dSrc.shape
           val begin = Array.fill(shape.length)(0)