提交

提交图

2937 次代码提交

作者 SHA1 备注 提交日期
xiang song(charlie.song) 246df1a8f4 update contributors (#3347)
Co-authored-by: Ubuntu <ubuntu@ip-172-31-2-66.ec2.internal>
2021-09-11 09:40:05 -07:00
Da Zheng 06f62b99a0 Update contributors (#3346)
* update.

* update.

* fix.

Co-authored-by: Zheng <dzzhen@88665a0acb11.ant.amazon.com>
2021-09-11 00:31:18 -07:00
esang 6454c791b8 [Bugfix] Fix bugs of farthest_point_sampler (#3327)
* fix start_idx

* fix the bug when cuda > 0

Co-authored-by: Tong He <hetong007@gmail.com>
2021-09-10 17:50:06 +08:00
Israt Nisa d17b299a13 [Feature] Update edge features (apply_edge) for heterogeneous graph (#3250)
* rgcn with new heterograph API

* apply_edge() forward for multi relation

* undoing changes from rgcn-hetero

* backward apply_edge(copy_u) added

* unittest for apply_edge(copy_e)

* Compatible with new PRs

* resolving conflict with master

* Bringing back change after resolving conflict

* minor

* minor

Co-authored-by: Israt Nisa <nisisrat@amazon.com>
2021-09-09 22:39:08 +08:00
Rhett Ying 129f67cd91 [Jenkins] add Rhett-Ying into authorized trigger (#3331) 2021-09-08 14:53:58 +08:00
Israt Nisa 298e4fa627 [Feature] Support builtin binary message function for heterogenenous graph (#3273)
* Added binary builtinMsgFunc forward() for heterograph

* Added backward for u_op_v

* Supports all binary builtin forward

* Supports binary message funcs with reduce func sum

* lint check

* removed import torch from unittest

* enabled GPU test

* lint check

* Fixed docstrings

* rename func get_hs_id

* edited comment

Co-authored-by: Israt Nisa <nisisrat@amazon.com>
2021-09-07 10:01:33 -07:00
Jinjing Zhou c81efdf215 Remove deprecated kernels (#3316)
* remove

* remove

* fix

* remove

* remove
2021-09-06 13:59:41 +08:00
Quan (Andy) Gan 75d793a1bc Update dataloader.py (#3322) 2021-09-06 12:16:03 +08:00
Tomasz Patejko f5183820d1 [Performance, CPU] Rewriting OpenMP pragmas into parallel_for (#3171)
* [CPU, Parallel] Rewriting omp pragmas with parallel_for

* [CPU, Parallel] Decrease number of calls to task function

* c[CPU, Parallel] Modify calls to new interface of parallel_for
2021-09-02 16:39:48 +08:00
xiang song(charlie.song) 21a4027969 Fix distributed device mapping problem. (#3313)
Co-authored-by: Ubuntu <ubuntu@ip-172-31-2-66.ec2.internal>
2021-09-02 13:52:00 +08:00
Rhett Ying 5a2451047f [Feature] enable to specify stream in UnitGraph::CopyTo() which could lead to async copy (#3297)
Co-authored-by: Minjie Wang <wmjlyjemaine@gmail.com>
2021-09-01 17:57:20 +08:00
xiang song(charlie.song) f4fe518ff7 [Feature] Add a HINT for the per edge type sampler of heterogeneous DistGraph that highlighting the etypes are sorted already. (#3260)
* pass cpp test

* distgraph use sorted edge flag.

* lint

* triger

* update test

Co-authored-by: Ubuntu <ubuntu@ip-172-31-2-66.ec2.internal>
2021-09-01 15:36:01 +08:00
nv-dlasalle 8e525dad71 [CPU][Sampling][Performance] Improve sampling on the CPU. (#3274)
* Optimize sampling

* Stop initialization of array

* Fix includes for linting

* Move comment

* Fix replace

Co-authored-by: Da Zheng <zhengda1936@gmail.com>
2021-08-31 13:37:49 -07:00
nv-dlasalle a53783c81d [Performance] Only copy and slice index's in subcolumn when data is used. (#2930)
* Make column double indices lazy

* Copy indices to proper contexts

* Fix initialization

* Add unit test

* Fix unit test for tensorflow

* Remove unused member

Co-authored-by: Quan (Andy) Gan <coin2028@hotmail.com>
2021-08-30 19:17:25 -07:00
Mufei Li 6476b829a8 [Doc] Add a Paper (#3301)
* Update README.md

* Update README.md
2021-08-30 16:11:55 +08:00
Da Zheng e5ed7adc12 [Distributed] Fix a bug in sampling an empty frontier (#3298)
* handle empty frontiers.

* fix lint.

* fix

Co-authored-by: Ubuntu <ubuntu@ip-172-31-2-202.us-west-1.compute.internal>
Co-authored-by: xiang song(charlie.song) <classicxsong@gmail.com>
2021-08-29 10:54:46 -07:00
NiklasBeierl 152760f66d [Doc] Note for HeteroGraphConv / Doc build problems (#3149)
* Doc: HetGConv note for `edge_type_subgraph`

For the rational behind this note see the discussion in #3003.

* WIP: Requirements for building the docs.

Had to install all these packages to get further with building the docs.

Didn't successfully build them yet, tho.

* Test

* Update

* Update hetero.py

* Update hetero.py

Co-authored-by: Ubuntu <ubuntu@ip-172-31-13-32.us-west-2.compute.internal>
Co-authored-by: Mufei Li <mufeili1996@gmail.com>
2021-08-29 03:14:43 +08:00
esang 4fb50be4d9 [Model] Point transformer (#3284)
* some modifications for pointnet2

* temporarily save changes

* move files to new directory point_transformer

* implement point transformer for classification

* restore train_cls in pointnet

* implement point transformer for partseg

* fix point transformer for nan loss

* modify point transformer for cls

* modify training setting

* update transformer for cls

* update code

* update code for latest performance

* update the example index

* some minor changes

Co-authored-by: Tong He <hetong007@gmail.com>
2021-08-27 14:33:18 +08:00
Konstantinos Vandikas 79305862ec allow for configuring default_dir (#3277)
* allow for configuring default_dir

* allow for using DGLDEFAULTDIR environment variable

* Update env_var.rst

Co-authored-by: Quan (Andy) Gan <coin2028@hotmail.com>
Co-authored-by: Jinjing Zhou <VoVAllen@users.noreply.github.com>
2021-08-27 01:29:49 +08:00
Rhett Ying 9cc85aed62 [Doc] fix formula in RelGraphConv (#3285)
* [Doc] fix formula in RelGraphConv

* refine descriptions
2021-08-25 17:59:17 +08:00
kalinin-sanja 4ef1c86b64 [Doc] Update 4_rgcn.py (#3261)
There is some misleading text.

Co-authored-by: Jinjing Zhou <VoVAllen@users.noreply.github.com>
Co-authored-by: Minjie Wang <wmjlyjemaine@gmail.com>
2021-08-25 15:56:29 +08:00
Quan (Andy) Gan 85b8fe5236 fix (#3286) 2021-08-24 14:22:49 +08:00
Quan (Andy) Gan b4cd60a952 fix relgraphconv bug (#3256) 2021-08-23 17:29:14 +08:00
Rhett Ying 8341244a2d [Doc] fix formula display issue in EdgeWeightNorm (#3275)
Co-authored-by: Quan (Andy) Gan <coin2028@hotmail.com>
2021-08-21 21:59:26 +08:00
Jinjing Zhou f67b1908ec add flag (#3264)
Co-authored-by: Quan (Andy) Gan <coin2028@hotmail.com>
2021-08-21 12:38:03 +08:00
Peiqi Yin f5b410b72d GeniePath model add a Tanh. (#3269)
* Update model.py

* Update README.md

Co-authored-by: Zihao Ye <expye@outlook.com>
2021-08-20 22:06:00 +08:00
nv-dlasalle 7f9279397a [Feature][DistDGL] Add NCCL support for range based partitions (#3213)
* Implement range based NDArrayPartition

* Finish implement range based partition support

* Add unit test

* Fix whitepace

* Add Kernel suffix

* Fix argument passing

* Add doxygen docs and improve variable naming

* Add unit test

* Add function for converting a partition book

* Add example to partition_op docs

* Fix dtype conversion for mxnet and tensorflow
2021-08-19 21:17:55 -07:00
Chen Kai 75fca8e4e8 [bugfix] fix the bug of DotGatConv when compute the edge_softmax (#3272)
Co-authored-by: Zihao Ye <expye@outlook.com>
2021-08-20 09:19:00 +08:00
nv-dlasalle f634950825 [Performance][Feature] Implement edge excluding in EdgeDataLoader on GPU (#3226)
* Update filter code

* Add unit tests

* Fixes

* Switch to indices

* Rename functions

* Fix linting

* Fix whitespace

* Add doc

* Fix heterograph

* Change workspace allocation

* Fix linting

* Fix docs in filter.py

* Add todo

Co-authored-by: Jinjing Zhou <VoVAllen@users.noreply.github.com>
Co-authored-by: Quan (Andy) Gan <coin2028@hotmail.com>
2021-08-19 12:24:15 -07:00
nv-dlasalle fc6f0b9efc [bugfix] Link against pytorch library by name rather than path (attempting to fix linking issue). (#3246)
* Use library name

* fix for mac builds from source

Co-authored-by: Quan (Andy) Gan <coin2028@hotmail.com>
2021-08-19 16:16:08 +08:00
maqy ea06688ed2 [Model] add model example EvolveGCN. (#3190)
* add evolveGCN example

* small fix

* fix defect

* fix defect

Co-authored-by: Quan (Andy) Gan <coin2028@hotmail.com>
2021-08-19 14:14:55 +08:00
Quan (Andy) Gan a536772ed2 fix cuda 11.1 crashing bug (#3265) 2021-08-18 21:16:49 +08:00
David Min 2613f7f013 [Performance] Cacheline-aligned access for UnifiedTensor (#3254)
* Add pytorch-direct version

* remove

* add documentation for UnifiedTensor

* Revert "add documentation for UnifiedTensor"

This reverts commit 63ba42644d4aba197c1cb4ea4b85fa1bc43b8849.

* alignment fix for UnifiedTensor access

* fix linting issue

Co-authored-by: shhssdm <shhssdm@gmail.com>
Co-authored-by: xiang song(charlie.song) <classicxsong@gmail.com>
2021-08-18 01:29:13 +08:00
Zihao Ye 76af2a2ef3 [perf] Remove activation cache if not required. (#3258)
* upd

* fix

* upd
2021-08-17 17:14:51 +08:00
Eric Kim ac01e88082 [Tools] In tools/launch.py, correctly pass all DGL client/server env vars if udf is a multi-command (#3245)
* Correctly pass all DGL client/server env vars if udf is a multi-command

* Refactor to use wrap_cmd_with_local_envvars() helper fn
2021-08-16 18:54:48 -07:00
Jinjing Zhou 1d4d4f59a6 Fix regression test (#3257) 2021-08-16 15:30:30 +08:00
Peiqi Yin 68c0cfbb3c [Model] Fix diffpool loss (#3233) 2021-08-16 15:18:15 +08:00
blokhinnv 31772b142e fixed typos in docs and docstring (#3231)
Co-authored-by: Quan (Andy) Gan <coin2028@hotmail.com>
2021-08-16 13:58:45 +08:00
xiang song(charlie.song) 6f36dd6366 Cleanup some code in graph_services.py (#3238)
* Fix bug

* Fix

* Fix

* upd

* Merge some code

* lint
2021-08-15 11:58:37 -07:00
Alex Morehead 1b350e9319 Fix small typo in knn_graph() documentation (#3244) 2021-08-12 14:23:27 +08:00
JOHNW02 738b75f41e [example] Create EEG-GCNN example. (#3186)
* Create EEG-GCNN example.

* Update README.md

* Remove gitignore file.

* Update README.md

* change 'datas' to 'datasets'.

* Change train.py to main.py

* Added an entry in the indexing page.

* State "simplified version"; change how to run.

* Fix bug in contact

* Remove paper link in reference.

* Create working branch

* Add normalization of x.

* Update paper link and tags

* Update paper link in readme

* Update readme; add patient level indices

* Update readme. Add comments to models

* Update README.md

* change to with; specify location for ch and el; move note

* fix bug for note

* Add args for models; clean code.

* delete = in readme

* Add reference for spec_coh_values
2021-08-11 13:47:59 +08:00
xiang song(charlie.song) cd2cf606db [Bugfix] Fix bug introduced by https://github.com/dmlc/dgl/pull/3131 (#3234)
* Fix bug

* Fix

* Fix

* upd

* trigger
2021-08-10 11:36:57 +08:00
nv-dlasalle 799c091e6d [bugfix] Only link the pytorch tensordispatcher against libtorch (#3225)
* Only link tensordispatcher against pytorch

* Only modify libraries when not using MSVC
2021-08-08 13:17:04 +08:00
xiang song(charlie.song) bcf92f6c21 [Bugfix] Distributed training can not work with dgl.dataloading.negative_sampler (#3215)
* Fix dist negative data loader bug

* upd

* Fix

Co-authored-by: Da Zheng <zhengda1936@gmail.com>
2021-08-06 20:41:20 +08:00
nv-dlasalle 4d9beb3b7b [Fix] FIx check for num_workers for using ScalarDataBatcher (#3219)
Co-authored-by: Quan (Andy) Gan <coin2028@hotmail.com>
Co-authored-by: Jinjing Zhou <VoVAllen@users.noreply.github.com>
2021-08-06 17:56:15 +08:00
Mufei Li 8c2f8ff84a Update citation_graph.py (#3221) 2021-08-06 15:57:42 +08:00
freeliuzc 45858d6803 [feature] add count_nonzero function for DistTensor (#3203)
* add count_nonzero function for DistTensor

* change the load method of local data

Co-authored-by: liuzichang04 <liuzichang04@meituan.com>
Co-authored-by: Jinjing Zhou <VoVAllen@users.noreply.github.com>
Co-authored-by: Da Zheng <zhengda1936@gmail.com>
2021-08-05 10:47:50 -07:00
Jinjing Zhou 8b5f4f5b1c Fix typos (#3214) 2021-08-03 17:39:45 +08:00
Yuchen 4ae13bd270 [bugfix] Fix force_reload parameter of FraudDataset (#3210)
* enable force_reload of FraudDataset

* rewrite hash_key in FraudDataset
2021-08-03 13:14:17 +08:00
Ankit Garg b131920066 [Feature] Add removed edges in distributed graph partitioning to handle heterogeneous graph (#3137)
* Added code for Rectifying (TypeError: unhashable type: 'slice') when copying file

* 1) added distributed preprocessing code to create ParMetis Input from CSV files
2) add code to run pm_dglpart on multiple machines
3) added support for recreating heteregenous graph from homo geneous graph based on dropped edges, as ParMetis currently only supports homogeneous graphs

* move to pandas

* Added comments and remove drop_duplicates as it was redundant

* Addressed Pr Comments

* Rename variable

* Added comment

* Added comment

* updated ReadMe

Co-authored-by: Ankit Garg <gaank@amazon.com>
Co-authored-by: Da Zheng <zhengda1936@gmail.com>
2021-08-02 11:41:13 -07:00