From: Nick Terrell <terrelln@fb.com>
Date: Tue, 10 Jan 2017 00:49:04 +0000 (-0800)
Subject: Use COVER_MEMMULT when training with COVER.
X-Git-Tag: v1.1.3^2~28^2~2
X-Git-Url: http://git.ipfire.org/cgi-bin/gitweb.cgi?a=commitdiff_plain;h=c220d4c74daf105b8425b3c7bfee4910f7286b8e;p=thirdparty%2Fzstd.git

Use COVER_MEMMULT when training with COVER.
---

diff --git a/programs/dibio.c b/programs/dibio.c
index 33c0250c5..5ef202c8a 100644
--- a/programs/dibio.c
+++ b/programs/dibio.c
@@ -235,7 +235,8 @@ int DiB_trainFromFiles(const char* dictFileName, unsigned maxDictSize,
     void* const dictBuffer = malloc(maxDictSize);
     size_t* const fileSizes = (size_t*)malloc(nbFiles * sizeof(size_t));
     unsigned long long const totalSizeToLoad = DiB_getTotalCappedFileSize(fileNamesTable, nbFiles);
-    size_t const maxMem =  DiB_findMaxMem(totalSizeToLoad * MEMMULT) / MEMMULT;
+    size_t const memMult = params ? MEMMULT : COVER_MEMMULT;
+    size_t const maxMem =  DiB_findMaxMem(totalSizeToLoad * memMult) / memMult;
     size_t benchedSize = (size_t) MIN ((unsigned long long)maxMem, totalSizeToLoad);
     void* const srcBuffer = malloc(benchedSize+NOISELENGTH);
     int result = 0;