if mxlen == 0, nothing to transfer

emer · Aug 21, 2020 · 8485fe9 · 8485fe9
1 parent 9d9334e
commit 8485fe9
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/empi/tensor.go b/empi/tensor.go
@@ -108,6 +108,9 @@ func GatherTensorRowsString(dest, src *etensor.String, comm *mpi.Comm) error {
 	for _, l := range dln {
 		mxlen = ints.MaxInt(mxlen, l)
 	}
+	if mxlen == 0 {
+		return nil // nothing to transfer
+	}
 	sdt := make([]byte, ssz*mxlen)
 	ddt := make([]byte, dsz*mxlen)
 	idx := 0