How dgl code finally converts to cuda and then runs on nvidia gpu? what is the flow it follows?
You may follow the guide for creating graphs on gpu. Any operations involving a gpu graph (including message passing in GNN) are performed on GPU. I’m not sure if this solves your question. If not, feel free to elaborate further.
This topic was automatically closed 30 days after the last reply. New replies are no longer allowed.