Skip to content

Commit 2748f6c

Browse files
committed
Reduce device size.
1 parent 231e4dd commit 2748f6c

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/data/batch_utils.cc

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -84,7 +84,7 @@ void CheckParam(BatchParam const& init, BatchParam const& param) {
8484
cache_host_ratio = 0.0;
8585
} else {
8686
// The number of bytes that must be in the host memory.
87-
auto h_cache_nbytes = n_cache_bytes - d_cache_nbytes;
87+
auto h_cache_nbytes = n_cache_bytes - d_cache_nbytes * 0.85;
8888
cache_host_ratio = static_cast<double>(h_cache_nbytes) / static_cast<double>(n_cache_bytes);
8989
if (lc > 0) {
9090
// 0 < lc < 10, C2C is available, but with reduced link count.

0 commit comments

Comments
 (0)