-
-
Notifications
You must be signed in to change notification settings - Fork 883
New issue
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 special functions from the CUDA Math API #6317
Add special functions from the CUDA Math API #6317
Conversation
In practice order is likely to be a very small integer, but it is annoying to users if they have to cast the first input to int32 rather than allowing the default Python int type.
I guess the |
This pull request is now in conflicts. Could you fix it @grlee77? 🙏 |
This pull request is now in conflicts. Could you fix it @grlee77? 🙏 |
/test mini |
Please test again (the prior failure in |
/test mini |
LGTM! Thanks! |
This is a small PR adding simple ufunc wrappers for a handful of other CUDA Math API functions that overlap with functions in
cupyx.scipy.special
.One question:
scipy.special.sinc
isnumpy.sinc
. This function promotes any integer dtype to double while the implementation here does not. For example, forcupy.sinc
as used here, providingcp.int8
input will result in acp.float16
array being returned. Should we make a wrapper that promotes any ints tocp.float64
for consistency with SciPy?