From b640bc99952d4541a6f43213f57883f2798d76b6 Mon Sep 17 00:00:00 2001 From: Eric Fu Date: Fri, 24 Nov 2023 10:10:34 +0000 Subject: [PATCH] increase_reserved_mem --- src/compute/src/memory/config.rs | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/compute/src/memory/config.rs b/src/compute/src/memory/config.rs index 186c41a021d70..3d959e19aa75b 100644 --- a/src/compute/src/memory/config.rs +++ b/src/compute/src/memory/config.rs @@ -157,9 +157,9 @@ mod tests { #[test] fn test_reserve_memory_bytes() { // at least 512 MB - let (reserved, non_reserved) = reserve_memory_bytes(1 << 30); + let (reserved, non_reserved) = reserve_memory_bytes(1536 << 20); assert_eq!(reserved, 512 << 20); - assert_eq!(non_reserved, 1536 << 20); + assert_eq!(non_reserved, 1024 << 20); // reserve based on proportion let (reserved, non_reserved) = reserve_memory_bytes(10 << 30);