Does DGL support M2 GPU acceleration?
Specifically, does model.to(‘mps’) and dgl.graph().to(‘mps’) works?
Encourage to move M2 related discussion to this github issue: Mac MPS support · Issue #4725 · dmlc/dgl · GitHub . Short answer is no as it requires new kernel implementation.