@@ -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
5050def _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