rapidsai / cuml

cuML - RAPIDS Machine Learning Library

Home Page:https://docs.rapids.ai/api/cuml/stable/

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Replace deprecated `cupyx.scatter_max` and `cupyx.scatter_add` with `cupy.maximum.at` and `cupy.add.at`

mroeschke opened this issue · comments

There are 3 usages of these methods in https://github.com/rapidsai/cuml/blob/branch-24.06/python/cuml/naive_bayes/naive_bayes.py

cupyx.scatter_max(

cupyx.scatter_max(highest_feature, indices[0], indices[2])

cupyx.scatter_add(jll[:, i], X.row, jll_data)

cupy recommends this being replaced by cupy.maximum.at and cupy.add.at.