search

SDCCA / DGL-PTM

The deep graph library implementation of the poverty trap model
GNU Lesser General Public License v2.1
0 stars 0 forks source link

Add mps capability #120

Open vmgaribay opened 2 months ago

vmgaribay commented 2 months ago

I would like to add mps device capability when I have a chance. I have done it for a nn training code a few months ago. I think I can apply the same here, but I need to give it actual attention.

vmgaribay-uva commented 2 months ago

Currently no support for DGL with MPS, https://github.com/dmlc/dgl/issues/7106