add comments for the tricky block matrix multiplication

apache · Sep 19, 2019 · f85cce1 · f85cce1
1 parent 2c37b58
commit f85cce1
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/operator/numpy/linalg/np_gesvd-inl.h b/src/operator/numpy/linalg/np_gesvd-inl.h
@@ -236,6 +236,7 @@ struct gesvd_backward {
 
     // G3:
     gemm::op(tempM, V, dA, DType(1.0), DType(1.0), false, false, s);
+    // dA <- dot(UT, dA). Loop over (k, m, m) blocks to avoid large temporary memory
     for (int i = 0; i < n; i += m) {
       int ncols = n - i < m ? n - i : m;
       Tensor<xpu, 3, DType> t = Tensor<xpu, 3, DType>(dA.dptr_ + i,