reid from https://github.com/michuanhaohao/reid-strong-baseline
zhangmeng
2020-01-17 1bb4d137919cae4f57f95a2572ee612dcabb3b3d
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
#pragma once
 
#include <c10/util/Logging.h>
#include <c10/util/Optional.h>
 
C10_DECLARE_bool(caffe2_cpu_numa_enabled);
 
namespace c10 {
 
/**
 * Check whether NUMA is enabled
 */
C10_API bool IsNUMAEnabled();
 
/**
 * Bind to a given NUMA node
 */
C10_API void NUMABind(int numa_node_id);
 
/**
 * Get the NUMA id for a given pointer `ptr`
 */
C10_API int GetNUMANode(const void* ptr);
 
/**
 * Get number of NUMA nodes
 */
C10_API int GetNumNUMANodes();
 
/**
 * Move the memory pointed to by `ptr` of a given size to another NUMA node
 */
C10_API void NUMAMove(void* ptr, size_t size, int numa_node_id);
 
/**
 * Get the current NUMA node id
 */
C10_API int GetCurrentNUMANode();
 
} // namespace c10