Using backend: pytorch
OGBg-molhiv dataset.
Traceback (most recent call last):
File "/home/jimmyzxj/miniconda3/envs/py39/lib/python3.9/runpy.py", line 197, in _run_module_as_main
return _run_code(code, main_globals, None,
File "/home/jimmyzxj/miniconda3/envs/py39/lib/python3.9/runpy.py", line 87, in _run_code
exec(code, run_globals)
File "/home/jimmyzxj/miniconda3/envs/py39/lib/python3.9/site-packages/memory_profiler.py", line 1349, in <module>
exec_with_profiler(script_filename, prof, args.backend, script_args)
File "/home/jimmyzxj/miniconda3/envs/py39/lib/python3.9/site-packages/memory_profiler.py", line 1250, in exec_with_profiler
exec(compile(f.read(), filename, 'exec'), ns, ns)
File "example.py", line 79, in <module>
main()
File "example.py", line 69, in main
g, task, datasets = prepare_dataset(*path_dict[task_name])
File "example.py", line 50, in prepare_dataset
g = glb.graph.read_glb_graph(metadata_path=metadata_path)
File "/home/jimmyzxj/glb/GLB-Repo/glb/graph.py", line 114, in read_glb_graph
return get_multi_graph(data, device)
File "/home/jimmyzxj/miniconda3/envs/py39/lib/python3.9/site-packages/memory_profiler.py", line 1186, in wrapper
val = prof(func)(*args, **kwargs)
File "/home/jimmyzxj/miniconda3/envs/py39/lib/python3.9/site-packages/memory_profiler.py", line 759, in f
return func(*args, **kwds)
File "/home/jimmyzxj/glb/GLB-Repo/glb/graph.py", line 82, in get_multi_graph
graphs.append(dgl.node_subgraph(g, node_list[i]))
File "/home/jimmyzxj/miniconda3/envs/py39/lib/python3.9/site-packages/dgl/subgraph.py", line 146, in node_subgraph
induced_nodes.append(_process_nodes(ntype, nids))
File "/home/jimmyzxj/miniconda3/envs/py39/lib/python3.9/site-packages/dgl/subgraph.py", line 139, in _process_nodes
return F.astype(F.nonzero_1d(F.copy_to(v, graph.device)), graph.idtype)
File "/home/jimmyzxj/miniconda3/envs/py39/lib/python3.9/site-packages/dgl/backend/pytorch/tensor.py", line 307, in nonzero_1d
x = th.nonzero(input, as_tuple=False).squeeze()
NotImplementedError: Could not run 'aten::nonzero' with arguments from the 'SparseCPU' backend. This could be because the operator doesn't exist for this backend, or was omitted during the selective/custom build process (if using custom build). If you are a Facebook employee using PyTorch on mobile, please visit https://fburl.com/ptmfixes for possible resolutions. 'aten::nonzero' is only available for these backends: [CPU, BackendSelect, Python, Named, Conjugate, Negative, ADInplaceOrView, AutogradOther, AutogradCPU, AutogradCUDA, AutogradXLA, AutogradLazy, AutogradXPU, AutogradMLC, AutogradHPU, AutogradNestedTensor, AutogradPrivateUse1, AutogradPrivateUse2, AutogradPrivateUse3, Tracer, UNKNOWN_TENSOR_TYPE_ID, Autocast, Batched, VmapMode].
CPU: registered at /opt/conda/conda-bld/pytorch_1640811723911/work/build/aten/src/ATen/RegisterCPU.cpp:18433 [kernel]
BackendSelect: fallthrough registered at /opt/conda/conda-bld/pytorch_1640811723911/work/aten/src/ATen/core/BackendSelectFallbackKernel.cpp:3 [backend fallback]
Python: registered at /opt/conda/conda-bld/pytorch_1640811723911/work/aten/src/ATen/core/PythonFallbackKernel.cpp:47 [backend fallback]
Named: registered at /opt/conda/conda-bld/pytorch_1640811723911/work/aten/src/ATen/core/NamedRegistrations.cpp:7 [backend fallback]
Conjugate: registered at /opt/conda/conda-bld/pytorch_1640811723911/work/aten/src/ATen/ConjugateFallback.cpp:18 [backend fallback]
Negative: registered at /opt/conda/conda-bld/pytorch_1640811723911/work/aten/src/ATen/native/NegateFallback.cpp:18 [backend fallback]
ADInplaceOrView: fallthrough registered at /opt/conda/conda-bld/pytorch_1640811723911/work/aten/src/ATen/core/VariableFallbackKernel.cpp:64 [backend fallback]
AutogradOther: registered at /opt/conda/conda-bld/pytorch_1640811723911/work/torch/csrc/autograd/generated/VariableType_0.cpp:8931 [autograd kernel]
AutogradCPU: registered at /opt/conda/conda-bld/pytorch_1640811723911/work/torch/csrc/autograd/generated/VariableType_0.cpp:8931 [autograd kernel]
AutogradCUDA: registered at /opt/conda/conda-bld/pytorch_1640811723911/work/torch/csrc/autograd/generated/VariableType_0.cpp:8931 [autograd kernel]
AutogradXLA: registered at /opt/conda/conda-bld/pytorch_1640811723911/work/torch/csrc/autograd/generated/VariableType_0.cpp:8931 [autograd kernel]
AutogradLazy: registered at /opt/conda/conda-bld/pytorch_1640811723911/work/torch/csrc/autograd/generated/VariableType_0.cpp:8931 [autograd kernel]
AutogradXPU: registered at /opt/conda/conda-bld/pytorch_1640811723911/work/torch/csrc/autograd/generated/VariableType_0.cpp:8931 [autograd kernel]
AutogradMLC: registered at /opt/conda/conda-bld/pytorch_1640811723911/work/torch/csrc/autograd/generated/VariableType_0.cpp:8931 [autograd kernel]
AutogradHPU: registered at /opt/conda/conda-bld/pytorch_1640811723911/work/torch/csrc/autograd/generated/VariableType_0.cpp:8931 [autograd kernel]
AutogradNestedTensor: registered at /opt/conda/conda-bld/pytorch_1640811723911/work/torch/csrc/autograd/generated/VariableType_0.cpp:8931 [autograd kernel]
AutogradPrivateUse1: registered at /opt/conda/conda-bld/pytorch_1640811723911/work/torch/csrc/autograd/generated/VariableType_0.cpp:8931 [autograd kernel]
AutogradPrivateUse2: registered at /opt/conda/conda-bld/pytorch_1640811723911/work/torch/csrc/autograd/generated/VariableType_0.cpp:8931 [autograd kernel]
AutogradPrivateUse3: registered at /opt/conda/conda-bld/pytorch_1640811723911/work/torch/csrc/autograd/generated/VariableType_0.cpp:8931 [autograd kernel]
Tracer: registered at /opt/conda/conda-bld/pytorch_1640811723911/work/torch/csrc/autograd/generated/TraceType_0.cpp:10285 [kernel]
UNKNOWN_TENSOR_TYPE_ID: fallthrough registered at /opt/conda/conda-bld/pytorch_1640811723911/work/aten/src/ATen/autocast_mode.cpp:466 [backend fallback]
Autocast: fallthrough registered at /opt/conda/conda-bld/pytorch_1640811723911/work/aten/src/ATen/autocast_mode.cpp:305 [backend fallback]
Batched: registered at /opt/conda/conda-bld/pytorch_1640811723911/work/aten/src/ATen/BatchingRegistrations.cpp:1016 [backend fallback]
VmapMode: fallthrough registered at /opt/conda/conda-bld/pytorch_1640811723911/work/aten/src/ATen/VmapModeRegistrations.cpp:33 [backend fallback]