leffff / graphormer-pyg

Microsoft Graphormer (https://arxiv.org/abs/2106.05234) rewritten in Pytorch-Geometric
MIT License
125 stars 17 forks source link

Parallelize Floyd-Warshall Algorithm with Multiprocessing #2

Open leffff opened 1 year ago

leffff commented 1 year ago

https://github.com/leffff/graphormer-pyg/blob/00e04752257e6e23d93cd6076bae66e687d013be/graphormer/functional.py#L74

leffff commented 1 year ago

Parallelisation commit delayed