We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
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
The text was updated successfully, but these errors were encountered:
@reduce
TODOs to get prange reductions for + and * working in main:
+
*
Sorry, something went wrong.
Fixed in #999
mingjie-intel
No branches or pull requests
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.
The text was updated successfully, but these errors were encountered: