Skip to content
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

Small enhancement to Cuda functions #1060

Merged
merged 4 commits into from
Apr 6, 2022

Conversation

hoxbro
Copy link
Member

@hoxbro hoxbro commented Apr 4, 2022

Adds Cuda support for category_binning and removes most of the following warning, the rest will be removed with #1016.

NumbaPerformanceWarning: Host array used in CUDA kernel will incur copy overhead to/from device.
    warn(NumbaPerformanceWarning(msg))

I'm not sure if I should replace cp.array with cp.asarray.

If needed, I can split this PR up into two.

Logfiles:
datashader_tests_cudf_21.10.01.log
datashader_tests_cudf_22.02.00.log

@philippjfr
Copy link
Member

I think cupy.asarray seems strictly better. I'm not sure if that codepath expects an xarray DataArray with a cupy array backing it but if it ever does then asarray will not make a copy.

Copy link
Member

@jbednar jbednar left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Looks great, thanks! I'd vote for asarray.

@jbednar jbednar merged commit d3fb6a3 into holoviz:master Apr 6, 2022
@jbednar
Copy link
Member

jbednar commented Apr 6, 2022

Thanks, @hoxbro!

@hoxbro hoxbro deleted the update_cuda_functions branch April 6, 2022 14:23
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants