Skip to content

Commit 1c4d3a1

Browse files
committed
[HOTFIX] Fixing chunking
1 parent 563127c commit 1c4d3a1

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

gempy_engine/modules/evaluator/generic_evaluator.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -20,8 +20,8 @@ def generic_evaluator(solver_input: SolverInput, weights: np.ndarray, options: I
2020
# * Chunking the evaluation
2121
max_size = options.evaluation_chunk_size
2222
n_chunks = int(np.ceil(matrix_size / max_size))
23-
chunk_size = int(np.ceil(matrix_size / n_chunks))
24-
for i in range(n_chunks):
23+
chunk_size = int(np.ceil(grid_size / n_chunks))
24+
for i in range(n_chunks): # TODO: It seems the chunking is not properly implemented
2525
slice_array = slice(i * chunk_size, (i + 1) * chunk_size)
2626
scalar_field_chunk, gx_field_chunk, gy_field_chunk, gz_field_chunk = _eval_on(
2727
solver_input=solver_input,
@@ -50,6 +50,7 @@ def generic_evaluator(solver_input: SolverInput, weights: np.ndarray, options: I
5050
def _eval_on(solver_input, weights, options, slice_array: slice = None):
5151
eval_kernel = yield_evaluation_kernel(solver_input, options.kernel_options, slice_array=slice_array)
5252
scalar_field: np.ndarray = (eval_kernel.T @ weights).reshape(-1)
53+
scalar_field[-50:]
5354
gx_field: Optional[np.ndarray] = None
5455
gy_field: Optional[np.ndarray] = None
5556
gz_field: Optional[np.ndarray] = None

0 commit comments

Comments
 (0)