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

[REVIEW]Add remote storage support #2859

Merged
Merged
Show file tree
Hide file tree
Changes from 45 commits
Commits
Show all changes
49 commits
Select commit Hold shift + click to select a range
3542708
Added the ability to load/call/unload general extensions in addition …
rlratzel Oct 26, 2022
2572ed0
Fixed tests broken by refactorings, removed dead code.
rlratzel Oct 26, 2022
203da90
Remove remove_pg_dependency_from_cugraph_store.py
VibhuJawa Oct 26, 2022
b19fab1
Added tests for extension modules that access the server and return N…
rlratzel Oct 26, 2022
e897caf
Merge remote-tracking branch 'upstream/branch-22.12' into branch-22.1…
rlratzel Oct 26, 2022
51fb38f
Remove remove_pg_dependency_from_cugraph_store.py
VibhuJawa Oct 26, 2022
d528c0c
Move code to helper functions to make them callable from extensions
VibhuJawa Oct 26, 2022
a18becd
Fix typos
VibhuJawa Oct 26, 2022
3b6ef59
Added copyrights
VibhuJawa Oct 26, 2022
4cfa293
Moved common stuff to RemoteStorage
VibhuJawa Oct 27, 2022
f1a3318
Address Reviews
VibhuJawa Oct 27, 2022
067c4e9
Added ucx-py infra for support result_device param for call_extension…
rlratzel Oct 27, 2022
ac59971
Added code to support loading extensions using a python module or pac…
rlratzel Oct 27, 2022
b4b489e
Updated docstring for load_extensions()
rlratzel Oct 27, 2022
943034f
Added Remote Storage Support
VibhuJawa Oct 27, 2022
82871c3
Added better error handling for APIs when using device transfer optio…
rlratzel Oct 27, 2022
42256e9
Merge remote-tracking branch 'upstream/branch-22.12' into branch-22.1…
rlratzel Oct 27, 2022
b0ffa45
Remove remove_pg_dependency_from_cugraph_store.py
VibhuJawa Oct 26, 2022
4df76f0
Remove remove_pg_dependency_from_cugraph_store.py
VibhuJawa Oct 26, 2022
a941a8a
Move code to helper functions to make them callable from extensions
VibhuJawa Oct 26, 2022
696fbc7
Fix typos
VibhuJawa Oct 26, 2022
8f9cfab
Added copyrights
VibhuJawa Oct 26, 2022
d73bef4
Moved common stuff to RemoteStorage
VibhuJawa Oct 27, 2022
e899c89
Address Reviews
VibhuJawa Oct 27, 2022
206e0be
Added Remote Storage Support
VibhuJawa Oct 27, 2022
0afbb9b
Add cugraph_service_store
VibhuJawa Oct 27, 2022
508f87a
fix merge conflicts
VibhuJawa Oct 27, 2022
8426656
Add RemotePropertyGraph and RemoteMGPropertyGraph
VibhuJawa Oct 27, 2022
bca4154
Merge commit 'refs/pull/2850/head' of https://github.com/rapidsai/cug…
VibhuJawa Oct 27, 2022
5e902e8
Added working feature storage and add parquet data
VibhuJawa Oct 28, 2022
00289a6
Add nvidia copyright
VibhuJawa Oct 28, 2022
5354648
Added copyright information
VibhuJawa Oct 28, 2022
5cdd07b
Added add_graph() to server facade so extensions can also create and …
rlratzel Oct 28, 2022
57fe0ea
Merge commit 'refs/pull/2850/head' of https://github.com/rapidsai/cug…
VibhuJawa Oct 28, 2022
133c0d0
add_remore_storage_support
VibhuJawa Oct 28, 2022
e91dc1f
Add LocalStore
VibhuJawa Oct 31, 2022
c0fab52
Added working graph service
VibhuJawa Oct 31, 2022
e55be6d
Merge branch 'branch-22.12' into add_remore_storage_support
VibhuJawa Nov 4, 2022
ce470bb
Style Fixes
VibhuJawa Nov 4, 2022
be2d7ca
Skipped pytest for now
VibhuJawa Nov 4, 2022
55ff270
Update python/cugraph/cugraph/gnn/dgl_extensions/cugraph_service_stor…
VibhuJawa Nov 7, 2022
56a4350
Address Reviews
VibhuJawa Nov 7, 2022
5b36168
Merge branch 'add_remore_storage_support' of https://github.com/Vibhu…
VibhuJawa Nov 7, 2022
d24ab52
Addresed using ntype and etype
VibhuJawa Nov 7, 2022
d7f4ace
fix typing issue
VibhuJawa Nov 7, 2022
00795b4
Update python/cugraph/cugraph/tests/test_dgl_extension_remote_wrapper…
VibhuJawa Nov 8, 2022
fdd39d3
remove relative imports
VibhuJawa Nov 8, 2022
35872e6
Merge branch 'add_remore_storage_support' of https://github.com/Vibhu…
VibhuJawa Nov 8, 2022
bf49417
link cugraph/issues/2863
VibhuJawa Nov 8, 2022
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
1 change: 1 addition & 0 deletions python/cugraph/cugraph/gnn/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -12,4 +12,5 @@
# limitations under the License.

from .dgl_extensions.cugraph_store import CuGraphStore
VibhuJawa marked this conversation as resolved.
Show resolved Hide resolved
from .dgl_extensions.cugraph_service_store import CuGraphRemoteStore
from .dgl_extensions.feature_storage import CuFeatureStorage
12 changes: 12 additions & 0 deletions python/cugraph/cugraph/gnn/dgl_extensions/__init__.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,12 @@
# Copyright (c) 2022, NVIDIA CORPORATION.
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
Loading