Thanks to visit codestin.com
Credit goes to github.com

Skip to content

Conversation

@puririshi98
Copy link
Contributor

@puririshi98 puririshi98 commented Jul 13, 2022

also includes backward for other in segment_matmul

puririshi98 and others added 30 commits July 12, 2022 17:44
puririshi98 and others added 19 commits July 22, 2022 15:24
@puririshi98
Copy link
Contributor Author

puririshi98 commented Jul 28, 2022

functional, added a test to ensure this https://github.com/puririshi98/pyg-lib/blob/back_for_grouped/test/ops/test_cutlass_autograd_ops.py
unfortunately the backward pass is super slow

@rusty1s rusty1s changed the title Autograd for grouped_matmul Autograd for grouped_matmul Aug 1, 2022
@rusty1s rusty1s merged commit 0a99d8f into pyg-team:master Aug 1, 2022
@rusty1s
Copy link
Member

rusty1s commented Aug 1, 2022

I have merged this now. We can continue working on it in future PRs.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants