pyg-team / pytorch_geometric

Graph Neural Network Library for PyTorch
https://pyg.org
MIT License
20.52k stars 3.57k forks source link

[ISSUE 9455] Data preprocess fix in multi XPU training #9456

Closed zhouyu5 closed 5 days ago

zhouyu5 commented 5 days ago

This is a follow-up PR for issue #9455 .

Say we want to lauch training on two XPUs: