From: Tom Clegg <tom@curii.com>
Date: Thu, 27 Jan 2022 14:16:35 +0000 (-0500)
Subject: Update memory-size log message.
X-Git-Url: https://git.arvados.org/lightning.git/commitdiff_plain/b9a352b3c94fbefd189bf72694ef52fe561f0515

Update memory-size log message.

refs #18581

Arvados-DCO-1.1-Signed-off-by: Tom Clegg <tom@curii.com>
---

diff --git a/slicenumpy.go b/slicenumpy.go
index 9e5e315c50..91413824fb 100644
--- a/slicenumpy.go
+++ b/slicenumpy.go
@@ -596,9 +596,10 @@ func (cmd *sliceNumpy) RunCommand(prog string, args []string, stdin io.Reader, s
 				throttleNumpyMem.Release()
 			}
 			if *onehotSingle {
-				log.Infof("%04d: keeping onehot chunk in memory (rows=%d, cols=%d, mem=%d)", infileIdx, len(cmd.cgnames), len(onehotChunk), (len(cmd.cgnames)+int(onehotXrefSize))*len(onehotChunk))
 				onehotIndirect[infileIdx] = onehotChunk2Indirect(onehotChunk)
 				onehotXrefs[infileIdx] = onehotXref
+				n := len(onehotIndirect[infileIdx][0])
+				log.Infof("%04d: keeping onehot coordinates in memory (n=%d, mem=%d)", infileIdx, n, n*8)
 			}
 			if !(*onehotSingle || *onehotChunked) || *mergeOutput || *hgvsSingle {
 				log.Infof("%04d: preparing numpy", infileIdx)