MCPcopy
hub / github.com/dmlc/dgl / pin_memory

Method pin_memory

python/dgl/graphbolt/base.py:499–502  ·  view source on GitHub ↗

Copy `SampledSubgraph` to the pinned memory using reflection.

(self)

Source from the content-addressed store, hash-verified

497 return self
498
499 def pin_memory(self):
500 """Copy `SampledSubgraph` to the pinned memory using reflection."""
501
502 return self.to("pinned")
503
504 def is_pinned(self) -> bool:
505 """Check whether `SampledSubgraph` is pinned using reflection."""

Callers 9

apply_toFunction · 0.45
_pinMethod · 0.45
toMethod · 0.45
test_pin_memoryFunction · 0.45
test_pin_unpinFunction · 0.45
test_CopyToFunction · 0.45
test_index_selectFunction · 0.45
test_index_select_cscFunction · 0.45
test_gpu_graph_cacheFunction · 0.45

Calls 1

toMethod · 0.95

Tested by 6

test_pin_memoryFunction · 0.36
test_pin_unpinFunction · 0.36
test_CopyToFunction · 0.36
test_index_selectFunction · 0.36
test_index_select_cscFunction · 0.36
test_gpu_graph_cacheFunction · 0.36