Skip to content

Numba Reduction operations (+=, *=) not supported on GPU #71

Closed
@adarshyoga

Description

@adarshyoga

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
No labels

Type

No type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions