OMP pragma was added to support hetero_graph loop. #2148
Merged
Conversation
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
mozga-intel
force-pushed
the
mozga-intel/parallel_hetero_graph_param_in_out
branch
2 times, most recently
from
2794db4
to
e3ac6b1
Compare
|
Could we only enable this optimization when the graph is stored on CPU? I don't think the optimization works for GPU as well. |
mozga-intel
force-pushed
the
mozga-intel/parallel_hetero_graph_param_in_out
branch
from
1105d3b
to
d624a99
Compare
|
@yzh119 Of course, done! |
mozga-intel
force-pushed
the
mozga-intel/parallel_hetero_graph_param_in_out
branch
from
d624a99
to
5a070b5
Compare
|
@yzh119 Thanks for your approval, could you please tell me whether it could be merged into the master branch? |
BarclayII
pushed a commit
to BarclayII/dgl
that referenced
this pull request
Nov 27, 2020
* OMP pragma was added to support hetero_graph loop, as follows: pragma omp parallel for * OMP pragma will be enabled iff the gpu is used for Co-authored-by: Zihao Ye <expye@outlook.com>
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
The OMP flag was added to support/obtain the inputs and outputs format of the matrices. This pull-request gives us the ability to get better performance in comparison with the master branch (approx. +5,6%). For brevity, the input and output format are triggered in a parallel way.