pyg-lib ToHeteroModule

[puririshi98]

benchmark of new HeteroModule:

sufficient acceleration achieved
Remaining TODO for the next PR:

• integrate into transformer setup

[codecov]

Codecov Report

Merging #5992 (a18c401) into master (08ce552) will increase coverage by 0.05%.
The diff coverage is 98.59%.

❗ Current head a18c401 differs from pull request most recent head 8f35fa4. Consider uploading reports for the commit 8f35fa4 to get more accurate results

@@            Coverage Diff             @@
##           master    #5992      +/-   ##
==========================================
+ Coverage   85.57%   85.62%   +0.05%     
==========================================
  Files         404      405       +1     
  Lines       22312    22383      +71     
==========================================
+ Hits        19094    19166      +72     
+ Misses       3218     3217       -1     

Impacted Files	Coverage Δ
torch_geometric/nn/to_hetero_module.py	98.59% <98.59%> (ø)
torch_geometric/nn/conv/hetero_conv.py	92.72% <0.00%> (+3.63%)	⬆️

📣 We’re building smart automated test selection to slash your CI/CD build times. Learn more

[rusty1s]

@puririshi98 I moved Linear and MessagePassing to its own classes instead of merging them two one. Hope that works for you :)