文件历史

128 次代码提交

作者 SHA1 备注 提交日期
Rhett Ying 03e83ac5a8 [dev] make sure graphbolt is not imported together with dgl (#7756) 2024-08-29 16:59:06 +08:00
Rhett Ying 00c50af3d7 [Dist] enable partition tests (#7675) 2024-08-08 16:32:33 +08:00
Rhett Ying 2ee8bc777b [CI] disable test_mp_dataloader due to glitch (#7490) 2024-06-27 13:39:51 +08:00
Rhett Ying 64750575a2 [CI] upgrade torch to 2.3.0 and cuda to 12.1 (#7399) 2024-05-11 16:21:05 +08:00
Rhett Ying 0de1de3dcd [dev] upgrade minimum required torch version to 2.0.0 (#7358) 2024-04-26 16:58:15 +08:00
Rhett Ying 0bfe34d96b [DistGB] refine convertion from dgl to graphbolt (#7007) 2024-01-25 14:55:25 +08:00
Rhett Ying cc1ebdf8b9 [dist] disable tool tests due to glitch 2024-01-02 15:02:54 +08:00
Rhett Ying f758c7c1d1 [DistDGL] fix distributed partition issue (#6847) 2023-12-30 07:48:59 +08:00
Rhett Ying ca675ed41d [GraphBolt] add torchdata into DGL dependencies (#6809) 2023-12-22 11:26:36 +08:00
Rhett Ying 19c6491b66 [DistGB] recover dist unit tests while skip unstable ones (#6582) 2023-11-21 16:44:10 +08:00
Hongzhi (Steve), Chen 346410928f [Misc] Cleanup old flags, and rely on BUILD_TYPE for all features. (#6154)
Co-authored-by: Ubuntu <ubuntu@ip-172-31-28-63.ap-northeast-1.compute.internal>
2023-08-15 16:31:44 +08:00
Hongzhi (Steve), Chen b9bb640a9a Revert "[Misc] clean up remove --inplace." (#6097) 2023-08-04 11:25:28 +08:00
Hongzhi (Steve), Chen 566719b149 [Misc] clean up remove --inplace. (#6096)
Co-authored-by: Ubuntu <ubuntu@ip-172-31-28-63.ap-northeast-1.compute.internal>
2023-08-04 09:06:05 +08:00
Hongzhi (Steve), Chen 3397496388 [Misc] Clean up build option for testing. (#6090)
Co-authored-by: Ubuntu <ubuntu@ip-172-31-28-63.ap-northeast-1.compute.internal>
2023-08-03 17:03:49 +08:00
Songqing Zhang b6f5ba9a80 [Misc] Fix the directory of DGL_DOWNLOAD_DIR (#6048) 2023-07-28 11:33:37 +08:00
Rhett Ying b26da25297 [GraphBolt] remove pydantic_yaml from dependency (#6011)
Co-authored-by: Hongzhi (Steve), Chen <chenhongzhi.nkcs@gmail.com>
2023-07-19 10:48:46 +08:00
Hongzhi (Steve), Chen 2746aac3ae [Misc] Use old pydantic yaml version for CI. (#6009)
Co-authored-by: Ubuntu <ubuntu@ip-172-31-28-63.ap-northeast-1.compute.internal>
2023-07-17 17:40:00 +08:00
Rhett Ying 39890c0c5d [GraphBolt] add pydantic-based metadata for TVT (#5942) 2023-07-04 20:55:26 +08:00
Rhett Ying 1cbe0b275e [CI] skip go tests 2023-07-01 10:40:11 +08:00
Rhett Ying 00972deecf [GraphBolt] convert dgl partition to CSCSamplingGraph (#5736) 2023-05-29 16:49:39 +08:00
peizhou001 494d2b7a07 [GraphBolt] Add graphbolt build into dgl build (#5696)
Co-authored-by: Ubuntu <ubuntu@ip-172-31-25-242.ap-northeast-1.compute.internal>
2023-05-16 08:58:31 +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
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
Hongzhi (Steve), Chen 325e795a18 Revert "Set USE_LIBXSMM default to OFF. (#5287)" (#5392)
This reverts commit a5e3139155.
2023-03-01 16:47:34 +08:00
Hongzhi (Steve), Chen 02f2526b85 [Misc] Fix typo in status.py (#5362)
Co-authored-by: Ubuntu <ubuntu@ip-172-31-28-63.ap-northeast-1.compute.internal>
2023-02-21 22:37:22 +08:00
Hongzhi (Steve), Chen 085b19d7c2 [Misc] Treat aborted as failure in Jenkins. (#5361)
Co-authored-by: Ubuntu <ubuntu@ip-172-31-28-63.ap-northeast-1.compute.internal>
2023-02-21 22:19:26 +08:00
Hongzhi (Steve), Chen e41ce0c66a [Misc] Update Jenkins status. (#5356)
* test

* blabla

* add

* reformat

* balbla

* rollback

* remove

---------

Co-authored-by: Ubuntu <ubuntu@ip-172-31-28-63.ap-northeast-1.compute.internal>
2023-02-21 18:45:00 +08:00
Hongzhi (Steve), Chen 71f7e96aeb [Misc] Abort previous CI. (#5346)
* Abort previous CI

* script

* newtry

* more

* fix

* comment

* add notes

---------

Co-authored-by: Ubuntu <ubuntu@ip-172-31-28-63.ap-northeast-1.compute.internal>
2023-02-21 15:09:30 +08:00
Hongzhi (Steve), Chen 27b3cd87e2 fix (#5349)
Co-authored-by: Ubuntu <ubuntu@ip-172-31-28-63.ap-northeast-1.compute.internal>
2023-02-21 14:02:25 +08:00
Hongzhi (Steve), Chen b8ed4d9ba3 [Misc] Revert status.py (#5348)
Co-authored-by: Ubuntu <ubuntu@ip-172-31-28-63.ap-northeast-1.compute.internal>
2023-02-21 14:00:08 +08:00
Hongzhi (Steve), Chen e4e0132972 [Misc] Add debug output to avoid silent failure. (#5347)
Co-authored-by: Ubuntu <ubuntu@ip-172-31-28-63.ap-northeast-1.compute.internal>
2023-02-21 13:42:00 +08:00
Hongzhi (Steve), Chen 0127a2ea4d [Misc] Update Jenkins status. (#5345)
* update

* polish

* polish

---------

Co-authored-by: Ubuntu <ubuntu@ip-172-31-28-63.ap-northeast-1.compute.internal>
2023-02-21 12:29:00 +08:00
Hongzhi (Steve), Chen e330954ca6 [Misc] Print jenkins response for test. (#5344)
Co-authored-by: Ubuntu <ubuntu@ip-172-31-28-63.ap-northeast-1.compute.internal>
2023-02-21 11:32:29 +08:00
Hongzhi (Steve), Chen deed945625 [Misc] Revert jenkins update change and add build url in log. (#5343)
Co-authored-by: Ubuntu <ubuntu@ip-172-31-28-63.ap-northeast-1.compute.internal>
2023-02-21 11:18:05 +08:00
Hongzhi (Steve), Chen f87e98e4b5 Update Jenkins response status (#5342)
Co-authored-by: Ubuntu <ubuntu@ip-172-31-28-63.ap-northeast-1.compute.internal>
2023-02-21 11:06:56 +08:00
Hongzhi (Steve), Chen 86193c26f7 [Misc] Reorg python test (with backend) according to the guideline. (#5312)
* rename

* remove test transpose

* udpate

* rename

* cleanup

---------

Co-authored-by: Ubuntu <ubuntu@ip-172-31-28-63.ap-northeast-1.compute.internal>
2023-02-17 16:11:39 +08:00
Hongzhi (Steve), Chen d9e550369f [Misc] Reorg python test according to the guideline. (#5307)
Co-authored-by: Ubuntu <ubuntu@ip-172-31-28-63.ap-northeast-1.compute.internal>
2023-02-17 16:08:20 +08:00
Hongzhi (Steve), Chen a5e3139155 Set USE_LIBXSMM default to OFF. (#5287)
* Set USE_LIBXSMM default to OFF.

* update

---------

Co-authored-by: Steve <ubuntu@ip-172-31-34-29.ap-northeast-1.compute.internal>
2023-02-16 13:08:11 +08:00
Quan (Andy) Gan fb8677b419 [Test] Move example test from unit test to examples (#5290)
* move example test from unit test to examples

* oops

* oops

* ok
2023-02-14 16:25:05 +08:00
Quan (Andy) Gan f62669b05c enable sparse on windows and mac (#5277)
* enable sparse on windows and mac

* that was stupid

* let's see what's going on..

* [Sparse] Fix the import error on Mac OS.

When using template functions that are defined in source files from DGL,
the loader of MacOS somehow cannot find their definitions. This fix simply
avoids depending on template functions from DGL headers.

With this fix, the sparse tests all pass on the MAC environment.

* ok this is the problem

* make errors clearer

* uh

* test

* Update __init__.py

* disabling ddp on windows

---------

Co-authored-by: czkkkkkk <zekucai@gmail.com>
2023-02-13 13:16:17 +08:00
Rhett Ying 465828cdbc [CI] pre-install required py packages in image for unit/tutorial test (#5276) 2023-02-10 19:11:30 +08:00
Rhett Ying 0c5d6f1b00 [BUILD] enable dgl sparse build in default (#5175)
* enable dgl sparse build in default

* disable sparse build on cugraph

* update
2023-01-16 14:01:39 +08:00
peizhou001 56ffb650fe [API Deprecation]Deprecate contrib module (#5114) 2023-01-06 20:16:01 +08:00
Minjie Wang 025e43210f [Sparse] Add tests for sparse examples; Migrate from mock_sparse to mock_sparse2 (#5073)
* pass gcn

* example tests

* pass gcnii, hgnn, sgc, sign

* black

* test file

* add torchmetrics

* migrate appnp, c_and_s, twirls

* fix style

* minor fix

* fix hypergraphatt and twirls

* pass gat

* fix torchmetric

* lintrunner

Co-authored-by: Mufei Li <mufeili1996@gmail.com>
2022-12-29 08:55:25 +08:00
Rhett Ying 9731e02325 [Dist] enable to chunk node/edge data into arbitrary number of chunks (#4930)
* [Dist] enable to chunk node/edge data into arbitrary number of chunks

* [Dist] enable to split node/edge data into arbitrary parts

* refine code

* Format boolean to uint8 forcely to avoid dist.scatter failure

* convert boolean to int8 before scatter and revert it after scatter

* refine code

* fix test

* refine code

* move test utilities into utils.py

* update comment

* fix empty data

* update

* update

* fix empty data issue

* release unnecessary mem

* release unnecessary mem

* release unnecessary mem

* release unnecessary mem

* release unnecessary mem

* remove unnecessary shuffle data

* separate array_split into standalone utility

* add example

Co-authored-by: xiang song(charlie.song) <classicxsong@gmail.com>
2022-12-15 10:01:39 +08:00
czkkkkkk 7c059e86c7 [Sparse] Add sparse matrix C++ implementation (#4773)
* [Sparse] Add sparse matrix C++ implementation

* Add documentation

* Update

* Minor fix

* Move Python code to dgl/mock_sparse2

* Move headers to include

* lint

* Update

* Add dgl_sparse directory

* Move src code to dgl_sparse

* Add __init__.py in tests to avoid naming conflict

* Add dgl sparse so in Jenkinsfile

* Complete docstring & SparseMatrix basic op

* lint

* Disable win tests
2022-11-07 18:01:19 +08:00
Xin Yao 96297fb8fd [Feature] Add bfloat16 (bf16) support (#4648)
* add bf16 specializations

* remove SWITCH_BITS

* enable amp for bf16

* remove SWITCH_BITS for cpu kernels

* enbale bf16 based on CUDART

* fix compiling for sm<80

* fix cpu build

* enable unit tests

* update doc

* disable test for CUDA < 11.0

* address comments

* address comments
2022-11-06 19:48:36 +08:00
Hongzhi (Steve), Chen ea48ce7a37 [Misc] Black auto fix. (#4697)
Co-authored-by: Steve <ubuntu@ip-172-31-34-29.ap-northeast-1.compute.internal>
2022-10-11 13:31:13 +08:00
Rhett Ying be11409312 [CI] Enable DGL cugaph nightly CI (#4573)
* [CI] Enable DGL cugaph nightly CI

* fix build issue

* remove runtime
2022-09-22 15:26:29 +08:00
Xin Yao 880b3b1f97 [Fix] Enable lint check for cuh files and fix compiler warnings (#4585)
* disable warning for tensorpipe

* fix warning

* enable lint check for cuh files

* resolve comments
2022-09-21 10:09:14 +08:00