Fix typo in non-distributed sampler

euro-hpc-pl · Sep 8, 2023 · c7729c3 · c7729c3
1 parent 1da7b48
commit c7729c3
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/omnisolver/bruteforce/gpu/sampler.py b/omnisolver/bruteforce/gpu/sampler.py
@@ -24,7 +24,7 @@ def sample(
         grid_size,
         block_size,
         num_steps_per_kernel=16,
-        partial_diff_buff_depth=1,
+        partial_diff_buffer_depth=1,
         dtype=np.float32,
     ):
         """Solve Binary Quadratic Model using exhaustive (bruteforce) search on the GPU.
@@ -49,7 +49,7 @@ def sample(
                 grid_size,
                 block_size,
                 num_steps_per_kernel,
-                partial_diff_buff_depth,
+                partial_diff_buffer_depth,
                 dtype,
             ).change_vartype("SPIN", inplace=False)
 
@@ -78,7 +78,7 @@ def sample(
                 block_size,
                 suffix_size,
                 num_steps_per_kernel,
-                partial_diff_buff_depth,
+                partial_diff_buffer_depth,
             )
         else:
             gpu_search(