Fix shmemalloc out of memory error

Author: hunhoffe

kernel/src/memory/shmemalloc.rs

@@ -33,17 +33,17 @@ unsafe impl Allocator for ShmemAlloc {
 
         let ptr = unsafe { alloc(layout) };
 
-        // TODO(rackscale performance): should probably be debug_assert
-        assert!(
-            is_shmem_addr(ptr as u64, true, true),
-            "allocated pointer ({}) isn't shmem: {} {} {}",
-            ptr as u64,
-            is_shmem_addr(ptr as u64, false, false),
-            is_shmem_addr(ptr as u64, false, true),
-            is_shmem_addr(ptr as u64, true, false)
-        );
-
         let ret = if !ptr.is_null() {
+            // TODO(rackscale performance): should probably be debug_assert
+            assert!(
+                is_shmem_addr(ptr as u64, true, true),
+                "allocated pointer ({}) isn't shmem: {} {} {}",
+                ptr as u64,
+                is_shmem_addr(ptr as u64, false, false),
+                is_shmem_addr(ptr as u64, false, true),
+                is_shmem_addr(ptr as u64, true, false)
+            );
+
             Ok(unsafe {
                 let nptr = NonNull::new_unchecked(ptr);
                 NonNull::slice_from_raw_parts(nptr, layout.size())

kernel/tests/s11_rackscale_benchmarks.rs

@@ -375,7 +375,7 @@ fn rackscale_vmops_benchmark(is_shmem: bool) {
             let timeout = 120_000 + 20000 * total_cores as u64;
 
             // TODO(rackscale): probably scale with nclients?
-            let shmem_size = SHMEM_SIZE;
+            let shmem_size = SHMEM_SIZE * 2;
             let all_outputs = Arc::new(Mutex::new(Vec::new()));
 
             // Run baseline test if needed