log intra-MKL memory use

techthiyanes · Aug 12, 2019 · ed0f3e2 · ed0f3e2
1 parent 1a48066
commit ed0f3e2
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/lenskit/algorithms/item_knn.py b/lenskit/algorithms/item_knn.py
@@ -380,7 +380,10 @@ def _mkl_similarities(self, rmat):
         m_nbrs = self.save_nbrs
         if m_nbrs is None or m_nbrs < 0:
             m_nbrs = 0
-        s_blocks = _mkl_sim_blocks(rmat.transpose().N, self.min_sim, m_nbrs)
+        rmat = rmat.transpose()
+        _logger.debug('[%s] transposed, memory use %s', self._timer, util.max_memory())
+        s_blocks = _mkl_sim_blocks(rmat.N, self.min_sim, m_nbrs)
+        _logger.debug('[%s] computed blocks, memory use %s', self._timer, util.max_memory())
         # blocks may be in any order, fix that
         s_blocks.sort(key=lambda b: b[0])
         s_blocks = [matrix.CSR(N=b) for (bi, b) in s_blocks]