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.

Sign up for GitHub

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

[REVIEW] Add make_meta_dispatch handling #637

Merged
merged 2 commits into from
Jun 3, 2021
Merged

[REVIEW] Add make_meta_dispatch handling #637

Changes from 1 commit
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
14d56d0
add make_meta handling
galipremsagar Jun 3, 2021
f0d4418
black
galipremsagar Jun 3, 2021
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
9 changes: 8 additions & 1 deletion dask_cuda/proxy_object.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -23,6 +23,13 @@
except ImportError:
from dask.dataframe.methods import concat_pandas

try:
from dask.dataframe.dispatch import (
make_meta_dispatch as make_meta_dispatch,
)
galipremsagar marked this conversation as resolved.
Show resolved Hide resolved
except ImportError:
from dask.dataframe.utils import make_meta as make_meta_dispatch

from .get_device_memory_objects import get_device_memory_objects
from .is_device_object import is_device_object

Expand Down Expand Up @@ -732,7 +739,7 @@ def wrapper(*args, **kwargs):
# Register dispatch of ProxyObject on all known dispatch objects
for dispatch in (
dask.dataframe.core.hash_object_dispatch,
dask.dataframe.utils.make_meta,
make_meta_dispatch,
dask.dataframe.utils.make_scalar,
dask.dataframe.core.group_split_dispatch,
dask.array.core.tensordot_lookup,
Expand Down