Unverified Commit b9c8d458 authored by Hongzhi (Steve), Chen's avatar Hongzhi (Steve), Chen Committed by GitHub
Browse files

move to cpu (#5187)


Co-authored-by: default avatarSteve <ubuntu@ip-172-31-34-29.ap-northeast-1.compute.internal>
parent f554bc48
......@@ -329,7 +329,7 @@
"val = torch.tensor([1., 2., 3., 4.])\n",
"A = dglsp.from_coo(row, col, val)\n",
"\n",
"B = A.to(device='cuda:0', dtype=torch.int32)\n",
"B = A.to(device='cpu', dtype=torch.int32)\n",
"print(f\"Device sparse matrix is stored on: {B.device}\")\n",
"print(f\"Datatype of sparse matrix: {B.dtype}\")"
],
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment