OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-07 12:21:27 +01:00
Code Issues Actions 38 Packages Projects Releases Wiki Activity
13,569 Commits 8,304 Branches 150 Tags 1.5 GiB
7122f8b3bb
Commit Graph

2 Commits

Author SHA1 Message Date
Dmytro Dzhulgakov
496c999f7d [core] NUMA-aware pinned allocator 
Using cudaHostRegister/Unregister instead of cudaMallocHost to move memory to a
specific NUMA node
 2018-03-06 00:33:11 -08:00
Dmytro Dzhulgakov
9e71de398b [core] Graph-level NUMA awareness in Caffe2 
Adding NUMA awareness through numa_node_id in DeviceOption. Blobs of operators
with numa_node_id are allocated on corr. memory banks, using CPU pools with
NUMA affinity set to run operators.
 2018-03-06 00:33:11 -08:00