benchmark fixes for lmdb/pi

misko · Mar 31, 2024 · 2eaa231 · 2eaa231
1 parent 03834e5
commit 2eaa231
Showing 1 changed file with 2 additions and 8 deletions.
diff --git a/spf/sdrpluto/benchmark.py b/spf/sdrpluto/benchmark.py
@@ -60,14 +60,8 @@ def benchmark(
         if store is None or store == "directory":
             store = zarr.DirectoryStore(filename)
         elif store == "lmdb":
-            store = zarr.LMDBStore(filename)
-            z = zarr.open(
-                filename,
-                mode="w",
-                shape=(2, total_samples),
-                chunks=(1, 1024 * chunk_size),
-                dtype="complex128",
-                compressor=compressor,
+            store = zarr.LMDBStore(
+                filename, map_size=2**40, writemap=True, map_async=True
             )
         else:
             raise NotImplementedError