mirrors / dmlc--dgl

提交

3,215 提交 34 分支 42 Git 标签
fix-gpu-sampling-doc
提交图

3215 次代码提交

此分支
此分支
所有分支
作者 SHA1 备注 提交日期
Ilia Taraban acb4eb7ecd [Feature] Add bfloat16 support for CPU (#5497) 
Co-authored-by: Hongzhi (Steve), Chen <chenhongzhi.nkcs@gmail.com>
 2023-04-06 11:16:41 +02:00
DominikaJedynak 29e6661523 [Optimization] Optimize bias term in GatConv layer (#5466) 2023-04-05 23:17:10 +08:00
Mufei Li a244de579c [Dataset] WisconsinDataset (#5520) 
Co-authored-by: Ubuntu <ubuntu@ip-172-31-36-188.ap-northeast-1.compute.internal>
 2023-04-04 16:44:26 +08:00
Ilia Taraban e8579d8be8 [Misc] Add MKL library path for standalone installation (#5510) 
Co-authored-by: Hongzhi (Steve), Chen <chenhongzhi.nkcs@gmail.com>
 2023-04-04 14:42:12 +08:00
Mufei Li dd4c74ff32 [Dataset] CornellDataset and TexasDataset (#5513) 
Co-authored-by: Ubuntu <ubuntu@ip-172-31-36-188.ap-northeast-1.compute.internal>
 2023-04-03 16:16:37 +08:00
Minjie Wang 10c4345648 [Doc] Update doc folder README (#5495) 2023-04-03 15:41:47 +08:00
czkkkkkk 4463b3d6a5 [Sparse] Support strided tensor when calling old DGL APIs (#5506) 2023-03-31 23:18:55 +08:00
Venzino.Han d45eafd47f [NN] add gatconv edge_weight (#5503) 
Co-authored-by: venzino-han <venzino.han@buzzvil.com>
Co-authored-by: Mufei Li <mufeili1996@gmail.com>
 2023-03-31 19:53:49 +08:00
Mufei Li 2b5921e759 [Dataset] Actor (#5511) 
Co-authored-by: Ubuntu <ubuntu@ip-172-31-36-188.ap-northeast-1.compute.internal>
 2023-03-31 10:28:50 +08:00
Hongzhi (Steve), Chen 5008af2210 [Misc] Rename number_of_edges and number_of_nodes to num_edges and num_nodes in examples. (#5492) 
* pytorch_example

* fix

---------

Co-authored-by: Ubuntu <ubuntu@ip-172-31-28-63.ap-northeast-1.compute.internal>
 2023-03-29 11:01:20 +08:00
Hongzhi (Steve), Chen 3c8ac0934e [Misc] Rename number_of_edges and number_of_nodes to num_edges and num_nodes. (#5490) 
* Other

* revert

---------

Co-authored-by: Ubuntu <ubuntu@ip-172-31-28-63.ap-northeast-1.compute.internal>
 2023-03-29 11:00:26 +08:00
Hongzhi (Steve), Chen 8f9f2e2a4d [Misc] Rename number_of_edges and number_of_nodes to num_edges and num_nodes. (#5472) 
* core

* revert

---------

Co-authored-by: Ubuntu <ubuntu@ip-172-31-28-63.ap-northeast-1.compute.internal>
 2023-03-29 08:57:47 +08:00
Mufei Li 690f37bbe9 update (#5507) 
Co-authored-by: Ubuntu <ubuntu@ip-172-31-36-188.ap-northeast-1.compute.internal>
 2023-03-28 19:22:44 +08:00
Hongzhi (Steve), Chen 89b321b847 [Misc] Rename number_of_edges and number_of_nodes to num_edges and num_nodes. (#5488) 
* nn-only

* data-only

---------

Co-authored-by: Ubuntu <ubuntu@ip-172-31-28-63.ap-northeast-1.compute.internal>
 2023-03-28 12:58:54 +08:00
Hongzhi (Steve), Chen 7f5da69709 [Misc] Rename number_of_edges and number_of_nodes to num_edges and num_nodes in dist related python files. (#5489) 
Co-authored-by: Ubuntu <ubuntu@ip-172-31-28-63.ap-northeast-1.compute.internal>
 2023-03-27 15:55:37 +08:00
Hongzhi (Steve), Chen 5b409bf7f5 Rename_test (#5487) 
Co-authored-by: Ubuntu <ubuntu@ip-172-31-28-63.ap-northeast-1.compute.internal>
 2023-03-24 18:44:24 +08:00
Hongzhi (Steve), Chen 0dd4f76745 Update Jenkinsfile (#5494) 2023-03-24 12:40:33 +08:00
Mufei Li 9e532e7def [Dataset] Chameleon (#5477) 
* update

* update

* update

* lint

* update

* CI

* lint

* update doc

---------

Co-authored-by: Ubuntu <ubuntu@ip-172-31-36-188.ap-northeast-1.compute.internal>
 2023-03-23 22:34:00 +08:00
Xin Yao 170203ae2b [Performance] Creating out buffers for segment_mm|sddmm via torch.empty() (#5462) 
* update for segmentMM

* update for sddmm

* fix a bug
 2023-03-23 17:41:07 +08:00
Hongzhi (Steve), Chen 0ac9bf34b5 Rename number_of_nodes, number_of_edges to num_nodes, num_edges. (#5486) 
* Add num_nodes/num_edges

* additional

* Update python/dgl/graph_index.py

Co-authored-by: Mufei Li <mufeili1996@gmail.com>

* Apply suggestions from code review

Co-authored-by: Mufei Li <mufeili1996@gmail.com>

---------

Co-authored-by: Ubuntu <ubuntu@ip-172-31-28-63.ap-northeast-1.compute.internal>
Co-authored-by: Mufei Li <mufeili1996@gmail.com>
 2023-03-23 17:30:45 +08:00
yangbaechu 097fd6dbb1 fixing typo in Korean User Guide (#5458) 2023-03-23 11:10:55 +08:00
Hongzhi (Steve), Chen 64a0c32d50 Authorize daniil-sizov, itaraban to trigger CI. (#5482) 2023-03-23 10:37:14 +08:00
Hongzhi (Steve), Chen 4785fc7565 Authorize bgawrych to trigger CI. 2023-03-23 09:02:07 +08:00
Mufei Li a07305301e Update main.py (#5460) 2023-03-22 21:01:16 +08:00
ZhenyuLU_Heliodore 4534650f5f [Dataset] Add ZINC Dataset (#5428) 
* Update dgl.data.rst

* Add files via upload

* Add files via upload

* Add files via upload

* Update zinc.py

* Update dgl.data.rst

* Update test_data.py

* Add files via upload

* Update cluster.py

* Update pattern.py

* Update zinc.py

* Update zinc.py

* Update test_data.py

* lint

* fix

* fix path

* update test on label shape

---------

Co-authored-by: rudongyu <ru_dongyu@outlook.com>
 2023-03-22 18:04:41 +08:00
Xin Yao acb955e15f [Cleanup] Cleanup unused CMake options (#5470) 
* cleanup unused cmake options

* disable BUILD_TORCH for cugraph

* resolve comments
 2023-03-22 16:06:44 +08:00
czkkkkkk 866c70dace [Sparse] Fix bugs in Diag format conversions (#5465) 2023-03-21 09:24:53 +08:00
Hongzhi (Steve), Chen 4cf5f682a6 [Misc] Polish metapath_randomwalk.h (#5471) 
Co-authored-by: Ubuntu <ubuntu@ip-172-31-28-63.ap-northeast-1.compute.internal>
 2023-03-20 22:17:39 +08:00
Hongzhi (Steve), Chen 8a83027274 [Misc] Merge test utils. (#5440) 
* merge

* format

* rename

* sort

* sort

* update

* update

* update

* Update tests/utils/checks.py

Co-authored-by: Mufei Li <mufeili1996@gmail.com>

---------

Co-authored-by: Ubuntu <ubuntu@ip-172-31-28-63.ap-northeast-1.compute.internal>
Co-authored-by: Mufei Li <mufeili1996@gmail.com>
 2023-03-20 15:32:45 +08:00
Hongzhi (Steve), Chen 5421940a77 [Misc] Fix C++ comment style. (#5467) 
Co-authored-by: Ubuntu <ubuntu@ip-172-31-28-63.ap-northeast-1.compute.internal>
 2023-03-20 10:26:53 +08:00
Theodore Vasiloudis 5cab4230fb [Dist] Add argument in dispatch_data.py to allow user-defined metadata JSON filename. (#5445) 2023-03-17 08:43:29 +08:00
yangbaechu 1e792886ed Fix typo (#5455) 2023-03-16 11:01:02 +08:00
Daniil Sizov 87fb7ed05b [Config] Enable libxsmm by default for AVX cpu (#5165) 
* Enable AVX by default

* Fix linting errors

* Fix win64 build (libxsmm not linked)

Libxsmm on Win64 is not linked, should be disabled by default

* Fix clang format issues

* Change lower supported cpu version to LIBXSMM_X86_AVX2

Change lower supported cpu version to LIBXSMM_X86_AVX2 to address https://github.com/dmlc/dgl/issues/3459 issue

* Fix unit test

Remove assumption that libxsmm is enabled in the config by default (only true for intel CPUs with AVX2 instructions)

---------

Co-authored-by: Ubuntu <ubuntu@ip-172-31-15-137.us-west-2.compute.internal>
Co-authored-by: Quan (Andy) Gan <coin2028@hotmail.com>
 2023-03-16 02:05:22 +08:00
Minjie Wang 8d99d30a82 add a link about the removed TGN example (#5456) 2023-03-15 22:08:38 +08:00
czkkkkkk 92e383c3b8 [Sparse] Remove the Python DiagMatrix class (#5444) 
* [Sparse] Remove the Python DiagMatrix class

* Update

* Update

* update

* Update
 2023-03-15 18:17:52 +08:00
Mufei Li f5ddb11448 Update neighbor_sampler.py (#5450) 2023-03-14 15:46:40 +08:00
kylasa 92e2299593 [DistDGL][Robustness]Replacing numpy's unique with custom implementation (#5391) 
* Replacing numpy's unique with custom implementation

* Added docstring to the new function.

* Adding unit tests

* Numpy's version issues with the 'kind' argument.

* Addressing CI Test Failure.

* Addressing CI review comments.

* revised implementation, optimized for time.

* added missing arguments for fallback case.

* Addressing CI test failures.

* Resolving issues with PYTHONPATH

* Fix CI Test Failure issues.

* fix CI test failures.

---------

Co-authored-by: Rhett Ying <85214957+Rhett-Ying@users.noreply.github.com>
 2023-03-10 13:38:27 -08:00
Rhett Ying d2ed57e94b [DistDGL] add log when load graph and data (#5350) 
* [DistDGL] add log when load graph and data

* fix lint

* refine log

---------

Co-authored-by: Israt Nisa <neesha295@gmail.com>
 2023-03-10 14:21:32 +08:00
kylasa e74b3d3dd2 [DistDGL][TestCoverage]Added testcase for testing distributed lookup service. (#5365) 
* Added testcase for testing distributed lookup service.

* Applying lintrunner patch.

* Fixing CI Test environment failures.

* lintrunner patch.

* lintrunner patch

* Fix CI Failure.

* Fixing CI Test failure cases.

* lintrunner patch.

* lintrunner patch and CI test failure.

* Restore no. of test cases.

* Resolving pythonpath issues.

* lintrunner patch.

* updating PYTHONPATH to resolve lib path

* Resolve merge conflicts

* Resolving issues with PYTHONPATH env variable.

* fix module path

* rename utils script under test to avoid ambiguity

* remove unnecessary pythonpath

* fix lint error

* fix lint error

---------

Co-authored-by: RhettYing <rhett_ying@qq.com>
 2023-03-10 12:19:41 +08:00
Gleb Bazhenov 54b4bd0acd [Data] new functional for creating data splits in graph (#5418) 
* new functional for creating data splits in graph

* minor fix in data split implementation

* apply suggestions from code review

Co-authored-by: Mufei Li <mufeili1996@gmail.com>

* refactoring + unit tests

* fix test file name

* move imports to the top

* Revert "fix test file name"

This reverts commit 126323e38c8de6fdc73c826dccc9048aed8f9634.

* remove nccl submodule

* address linter issues

---------

Co-authored-by: Mufei Li <mufeili1996@gmail.com>
Co-authored-by: Hongzhi (Steve), Chen <chenhongzhi.nkcs@gmail.com>
 2023-03-09 18:55:25 +08:00
czkkkkkk a03dec059e [Sparse] Support Diag sparse format in C++ (#5432) 
* [Sparse] Support Diag sparse format in C++

* update

* Update
 2023-03-09 12:49:47 +08:00
DominikaJedynak b7ce4b6a0c Tensorflow examples fix (#5433) 2023-03-09 10:38:19 +08:00
Minjie Wang e102543b5f Update stale.yml (#5441) 2023-03-09 10:21:21 +08:00
Hongzhi (Steve), Chen e250ba3d99 [Misc] Rename test_utils -> pytests_utils. (#5438) 
* rename

* sort

* rename

---------

Co-authored-by: Ubuntu <ubuntu@ip-172-31-28-63.ap-northeast-1.compute.internal>
 2023-03-09 09:42:29 +08:00
kkranen fe3d29ac4d [Dataloading] Ignore edge types without reverse types in edge dataloader (#5411) 
* Bugfix for reverse edge issue.

* Resolved copy/paste transcription error

* lint

* Added unit tests for graph with not all reverses.

* linting + added backend device test

---------

Co-authored-by: Quan (Andy) Gan <coin2028@hotmail.com>
Co-authored-by: Rhett Ying <85214957+Rhett-Ying@users.noreply.github.com>
 2023-03-08 22:04:51 +08:00
Xin Yao ae4a5b7394 [Feature] Add state_dict, load_state_dict, param_groups to dgl.optim.SparseGradOptimizer (#5311) 
* init update

* all get/set optm_state

* add unit tests

* add docstring

* fix for multiple embeddings

* move embedding methods to private

* fix lint

* fix unit tests

* resolve comments

* merge master
 2023-03-08 16:46:34 +08:00
Guangyu Zhou 178290249e [Dataset] Add PATTERN dataset (#5422) 
* add PATTERN dataset

* fix bug

* fix bugs

* fix issues

* refine according to dongyu's comments

---------

Co-authored-by: BuptTab <gyzhou2000@gmail.com>
Co-authored-by: rudongyu <ru_dongyu@outlook.com>
 2023-03-08 14:08:12 +08:00
Hongzhi (Steve), Chen a454734f16 [Misc] Follow up on the missing comment in PR#4989. (#5437) 
Co-authored-by: Ubuntu <ubuntu@ip-172-31-28-63.ap-northeast-1.compute.internal>
Co-authored-by: Xin Yao <yaox12@outlook.com>
 2023-03-08 13:32:21 +08:00
Xin Yao 8d5d8962ad [Refactor] Replace third_party/nccl with PyTorch's NCCL backend (#4989) 
* expose GeneratePermutation

* add sparse_all_to_all_push

* add sparse_all_to_all_pull

* add unit test

* handle world_size=1

* remove python nccl wrapper

* remove the nccl dependency

* use pinned memory to speedup D2H copy

* fix lint

* resolve comments

* fix lint

* fix ut

* resolve comments
 2023-03-08 12:59:10 +08:00
Rhett Ying b1ec112eeb Fix compile error on ubuntu22.04_g++11.3.0 (#5434) 2023-03-08 09:49:23 +08:00