Use correct number of bytes per double

bob-carpenter · Jul 8, 2024 · 0f704eb · 0f704eb
1 parent 2b773ed
commit 0f704eb
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/workflow/rules/0_common.smk b/workflow/rules/0_common.smk
@@ -23,7 +23,7 @@ def make_target_configs(config):
 
 
 def mem_mb_for_samples(wildcards) -> int:
-    bytes_per_double = 8
+    bytes_per_double = 64
     num_chains = config["sample"]["chains"]
     num_iters = config["sample"]["iter_sampling"]
     target_config = f"{wildcards.target}/{wildcards.target_config}"

diff --git a/workflow/rules/2_compute_target_expectations.smk b/workflow/rules/2_compute_target_expectations.smk
@@ -1,5 +1,5 @@
 def mem_mb_for_target_expectations(wildcards) -> int:
-    bytes_per_double = 8
+    bytes_per_double = 64
     scale_factor = 10
     batch_size = 1_000_000
     N = target_configs[wildcards.target_config][1]["N"]