search

CommediaJW / dgs-bench

0 stars 0 forks source link

Free memory space in time #3

Closed gpzlx1 closed 2 years ago

gpzlx1 commented 2 years ago

https://github.com/CommediaJW/dgs-bench/blob/338ef584453b9954a476b9630a4a252c4978e914/graphsage/shared_graph.py#L50

indptr = create_shared_tensor(indptr) 

# or
shared_indptr = create_shared_tensor(indptr)
del indptr
gpzlx1 commented 2 years ago

image

# arrow 2
features = graph.pop('feature')
labels = graph.pop('labels')

# arrow 3 
g.ndata.clear()
g.edata.clear()