mirrors / dmlc--dgl

提交

2,937 提交 34 分支 42 Git 标签
test_sparse_compile
提交图

2937 次代码提交

此分支
此分支
所有分支
作者 SHA1 备注 提交日期
nv-dlasalle 2bbca12ad5 [bugfix] Fix assertions in /src/runtime/workspace.h and expand unit tests for sparse optimizer (#5299) 
* Fix assertions for size 0 workspaces

* Expand unit test to cover case of communication

* Fixes

* Format

* Fix c++ formatting
 2023-02-16 13:43:03 +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
Hongzhi (Steve), Chen 0552466c9c sort (#5281) 
Co-authored-by: Steve <ubuntu@ip-172-31-34-29.ap-northeast-1.compute.internal>
 2023-02-16 13:06:46 +08:00
Hongzhi (Steve), Chen f7a0c4bbf9 remove (#5280) 
Co-authored-by: Steve <ubuntu@ip-172-31-34-29.ap-northeast-1.compute.internal>
 2023-02-16 12:57:49 +08:00
Tingyu Wang 19b3cea166 [Model] Update CuGraphRelGraphConv to use pylibcugraphops=23.02 (#5217) 
* update cugraph_relgraphconv

* update equality test

* update cugraph rgcn example

* update RelGraphConvAgg based on latest API changes

* enable fallback option to fg when fanout is large

---------

Co-authored-by: Mufei Li <mufeili1996@gmail.com>
 2023-02-15 20:11:20 +08:00
Quan (Andy) Gan 5f1babf0bf fix build for multiple pytorch versions (#5289) 2023-02-15 15:22:50 +08:00
Hongzhi (Steve), Chen 40c968b78b fix (#5293) 
Co-authored-by: Steve <ubuntu@ip-172-31-34-29.ap-northeast-1.compute.internal>
 2023-02-14 17:18:45 +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
Hongzhi (Steve), Chen 267d285597 [Dev] Create canonical dev conda environment. (#5278) 
* create_dev_conda_env

* fix

* polish

* nlabla

* cleanup

* more

* add_more

* polish

---------

Co-authored-by: Steve <ubuntu@ip-172-31-34-29.ap-northeast-1.compute.internal>
 2023-02-14 16:07:07 +08:00
kylasa 432c71ef25 Code changes to fix order sensitivity of the pipeline (#5288) 
Following changes are made in this PR.
1. In dataset_utils.py, when reading edges from disk we follow the order defined by the STR_EDGE_TYPE key in the metadata.json file. This order is implicitly used to assign edgeid to edge types. This same order is used to read edges from the disk as well.
2. Now the unit test framework will also randomize the order of edges read from the disk. This is done for the edges when reading from the disk for the unit tests.

Co-authored-by: Quan (Andy) Gan <coin2028@hotmail.com>
 2023-02-13 10:56:26 -08:00
Krzysztof Sadowski f5afc6eaa2 fix rounding issue (#5159) 
Co-authored-by: Mufei Li <mufeili1996@gmail.com>
 2023-02-13 18:02:55 +08:00
Zahi Kakish 21733e903b Documentation equation syntax fix for EGATConv (#5284) 
Some weird formatting occurred in the documentation for EGATConv, so this quick commit should fix that.

Co-authored-by: Mufei Li <mufeili1996@gmail.com>
 2023-02-13 14:58:11 +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
Nicolas Castet af79e54049 [Bugfix] Fix Column.record_stream(...) for unmaterialized tensors (#5240) 
* Fix Column.record_stream(...) for unmaterialized tensors

* lint

* Record streams for indices on non-materialized columns

* add docstring

* fix for cpu index

* Record also stream on storage

* Always call self.storage.record_stream when storage is on GPU

* fix lint

---------

Co-authored-by: Xin Yao <xiny@nvidia.com>
 2023-02-10 14:36:46 +08:00
Rhett Ying d49a3019de [Doc] update --part_config 2023-02-10 11:18:02 +08:00
Rhett Ying ed3888dd74 [Doc] update --part_config 2023-02-10 11:17:00 +08:00
peizhou001 f0b7cc96bd [Performance]Add concurrent cpu id hashmap (#5241) 
Add Id hash map
 2023-02-09 08:51:08 +08:00
ZhenyuLU_Heliodore 5598503aa0 [Transform] Add SVDPE Transform Module (#5121) 
* add SVD positional encoding

* modify importing module

* Fixed certain problems

* Change the test unit to a nonsigular one

* Fixed typo and make accord with lintrunner

* added svd_pe into dgl.rst

* Modified dgl.rst
 2023-02-08 12:29:03 +08:00
kylasa aa42aaeb9f [DistDGL][Lintrunner]Lintrunner for tools directory (#5261) 
* lintrunner patch for gloo_wrapper.py

* lintrunner changes to the tools directory.
 2023-02-03 09:56:47 -08:00
Hongzhi (Steve), Chen 8b47bad5ba [Misc] Remove unnecessary clean up before build to accelerate incremental build. (#5267) 
* remove unnecessary clean up.

* more

---------

Co-authored-by: Steve <ubuntu@ip-172-31-34-29.ap-northeast-1.compute.internal>
 2023-02-03 15:11:16 +08:00
Rhett Ying eff16b6116 [Dist] add input_dir for parmetis preprocess (#5232) 
* [Dist] add input_dir for parmetis preprocess

* add support for parquet

* update parmetis_wrapper accordingly
 2023-02-02 09:44:35 +08:00
Rhett Ying cf67829998 [Doc] fix docstring issue (#5260) 2023-02-01 16:13:43 +08:00
Hongzhi (Steve), Chen 77bcf6040e Update pip install info for tutorials. (#5236) 
* update pip install info.

* update

* clarify the version

* blabla

* blabl

---------

Co-authored-by: Steve <ubuntu@ip-172-31-34-29.ap-northeast-1.compute.internal>
 2023-02-01 15:17:40 +08:00
Hongzhi (Steve), Chen b3308ae795 Cleanup mock_sparse related code. (#5230) 
* remove_mock_sparse_example

* mock_sparse_test

* remove_mock_sparse

---------

Co-authored-by: Steve <ubuntu@ip-172-31-34-29.ap-northeast-1.compute.internal>
 2023-02-01 15:17:14 +08:00
Minjie Wang 829ce109f1 Update hgnn.ipynb (create by colab) 2023-01-30 20:14:01 +08:00
Rhett Ying 389b8eb928 [DOC] replace emoji with Y/N to fix display issue (#5235) 2023-01-30 15:14:19 +08:00
Minjie Wang 8230640248 Update index.rst (#5233) 2023-01-30 14:15:52 +08:00
Muhammed Fatih BALIN 5888d9d1b6 [Doc] Labor sampling improvement (#5221) 
* [Doc] Labor sampling improvement

When importance sampling option is used, the returned blocks include edge weights that need to be used during the message passing operation. So, I am including this in the documentation.

* Add documentation to weighted sampling case also
 2023-01-29 21:32:09 +08:00
Hongzhi (Steve), Chen 4b5fa83bcd [Sparse] Swap examples of from_coo and spmatrix. (#5228) 
* swap-example

* size-shape

Co-authored-by: Steve <ubuntu@ip-172-31-34-29.ap-northeast-1.compute.internal>
 2023-01-27 10:51:57 +08:00
Hongzhi (Steve), Chen 387ae76df3 [Doc] Update with Mono font to keep doc string consistent. (#5213) 
* part1

* more

* more

* spmatrix

* code

Co-authored-by: Steve <ubuntu@ip-172-31-34-29.ap-northeast-1.compute.internal>
 2023-01-23 09:48:07 +08:00
Hongzhi (Steve), Chen d61012e094 Change example code for indices creation from two line to one line to avoid test break. (#5222) 
* add_backslash

* fix

* more

* blabla

Co-authored-by: Steve <ubuntu@ip-172-31-34-29.ap-northeast-1.compute.internal>
 2023-01-23 02:21:57 +08:00
Minjie Wang 0bdc437b52 [Doc] Update colab tutorials with spmatrix (#5220) 
* Created using Colaboratory

* Created using Colaboratory

* Created using Colaboratory

* Created using Colaboratory

* Created using Colaboratory

* Created using Colaboratory
 2023-01-20 17:44:58 +08:00
Hongzhi (Steve), Chen fbbe6d61c9 [Doc] Update all canonical creation example from from_coo to spmatrix. (#5208) 
* fix

* sddmm

* change

* blabla

* blabla

* revert

* newpr

* tocuda

Co-authored-by: Steve <ubuntu@ip-172-31-34-29.ap-northeast-1.compute.internal>
 2023-01-20 17:44:07 +08:00
Hongzhi (Steve), Chen 30fb03a6bc Update example with spmatrix. (#5219) 
* update

* graph

Co-authored-by: Steve <ubuntu@ip-172-31-34-29.ap-northeast-1.compute.internal>
 2023-01-20 15:39:55 +08:00
Quan (Andy) Gan abfc4c31f5 [Sparse] Clarification on dimensions for sparse reduction and softmax docstrings (#5197) 
* fix sparse reduce and softmax doc

* Update softmax.py
 2023-01-20 14:31:33 +08:00
Augusto Berndt 94d3c6cb9f [Doc] Fix a typo in dgl.dataloading.rst (#5211) 
changing privimitive to primitive
 2023-01-20 14:23:26 +08:00
Hongzhi (Steve), Chen 7a2dd5ae96 32 (#5216) 
Co-authored-by: Steve <ubuntu@ip-172-31-34-29.ap-northeast-1.compute.internal>
 2023-01-20 11:26:26 +08:00
Hongzhi (Steve), Chen 17fad103b5 update (#5215) 
Co-authored-by: Steve <ubuntu@ip-172-31-34-29.ap-northeast-1.compute.internal>
 2023-01-20 11:25:53 +08:00
Rhett Ying a844d16ec0 [Doc] add sphinxemoji 2023-01-20 10:58:22 +08:00
Hongzhi (Steve), Chen c89ee05f2d examples (#5214) 
Co-authored-by: Steve <ubuntu@ip-172-31-34-29.ap-northeast-1.compute.internal>
 2023-01-20 10:46:12 +08:00
Rhett Ying 5337d7b2bc [Doc] fix math and link display issue (#5212) 2023-01-20 10:45:11 +08:00
czkkkkkk 20ce9be2c6 [Sparse] Add spmatrix creation function (#5209) 
* [Sparse] Add spmatrix creation function

* Update

* Update sparse_matrix.py

* Update sparse_matrix.py

Co-authored-by: Hongzhi (Steve), Chen <chenhongzhi.nkcs@gmail.com>
 2023-01-20 09:49:49 +08:00
Rhett Ying 6d28052479 [Doc] revert change in link 2023-01-20 08:42:12 +08:00
Rhett Ying cccbb9129a [Doc] revert ** removal from table 2023-01-20 08:41:06 +08:00
Rhett Ying 6324351b4f [Doc] fix link and table display issue 2023-01-20 08:35:55 +08:00
czkkkkkk d1e5700676 [Sparse] Polish the docstring of sparse_matrix.py (#5202) 
* [Sparse] Polish the docstring of sparse_matrix.py

* Update

* Update

* Update

* Update
 2023-01-20 00:49:20 +08:00
Quan (Andy) Gan f145097f9a [Release] Conda/CMake changes for 1.0 release (#5186) 
* update conda branch

* fix conda

* update conda

* update

* update windows build script

* copy static library to build dir

* try again

* fix

* lint
 2023-01-19 22:28:04 +08:00
Quan (Andy) Gan c134119096 [Sparse] Update HGNN notebook (#5206) 
* Created using Colaboratory

* Created using Colaboratory
 2023-01-19 22:27:57 +08:00
czkkkkkk db77283130 [Sparse] Only support scalar or vector values in SparseMatrix creators (#5204) 
* [Sparse] Only support scalar or vector values in SparseMatrix creators

* Update

* Update sparse_matrix.py

Co-authored-by: Hongzhi (Steve), Chen <chenhongzhi.nkcs@gmail.com>
 2023-01-19 21:46:46 +08:00