Closed
Description
Add-assign, multiply-assign operations within Numba par-fors which are reduction operations need to be supported on GPU.
Here is an example kernel in pygbm where "+=" reduction operation is necessary.
@njit(parallel=True)
def _update_raw_predictions(leaves_data, raw_predictions):
for leaf_idx in prange(len(leaves_data)):
leaf_value, sample_indices = leaves_data[leaf_idx]
for sample_idx in sample_indices:
raw_predictions[sample_idx] += leaf_value
Metadata
Metadata
Assignees
Labels
No labels