`segment_matmul` integration from `pyg-lib`
See original GitHub issue🚀 The feature, motivation and pitch
pyg-lib
issue: https://github.com/pyg-team/pyg-lib/issues/49pyg-lib
PR: https://github.com/pyg-team/pyg-lib/pull/51
Need to integrate this into the backend of PyG’s heterogeneous GNNs.
Issue Analytics
- State:
- Created a year ago
- Reactions:1
- Comments:6 (6 by maintainers)
Top Results From Across the Web
Haicheng Wu on LinkedIn: `cutlass` integration + ...
CUTLASS is integrated into PaddlePaddle to use fused input gather and output scatter GEMM kernels as a new way to do sparse.
Read more >
Top Related Medium Post
No results found
Top Related StackOverflow Question
No results found
Troubleshoot Live Code
Lightrun enables developers to add logs, metrics and snapshots to live code - no restarts or redeploys required.
Start Free
Top Related Reddit Thread
No results found
Top Related Hackernoon Post
No results found
Top Related Tweet
No results found
Top Related Dev.to Post
No results found
Top Related Hashnode Post
No results found
Important to note, this will require adding backward support of
segment_matmul
inpyg-lib
.heterolinear cutlass support: https://github.com/pyg-team/pytorch_geometric/pull/5330