doxygen-c/html/c10_2util_2numa_8h_source.html

 #pragma once

 #include <c10/util/Logging.h>
 #include <c10/util/Optional.h>

 C10_DECLARE_bool(caffe2_cpu_numa_enabled);

 namespace c10 {

 C10_API bool IsNUMAEnabled();

 C10_API void NUMABind(int numa_node_id);

 C10_API int GetNUMANode(const void* ptr);

 C10_API int GetNumNUMANodes();

 C10_API void NUMAMove(void* ptr, size_t size, int numa_node_id);

 C10_API int GetCurrentNUMANode();

 } // namespace c10
c10::NUMAMove
void NUMAMove(void *ptr, size_t size, int numa_node_id)
Move the memory pointed to by ptr of a given size to another NUMA node.
Definition: numa.cpp:123

c10::GetCurrentNUMANode
int GetCurrentNUMANode()
Get the current NUMA node id.
Definition: numa.cpp:126

c10::GetNUMANode
int GetNUMANode(const void *ptr)
Get the NUMA id for a given pointer ptr
Definition: numa.cpp:115

c10::IsNUMAEnabled
bool IsNUMAEnabled()
Check whether NUMA is enabled.
Definition: numa.cpp:108

c10
To register your own kernel for an operator, do in one (!) cpp file: C10_REGISTER_KERNEL(OperatorHand...
Definition: alias_info.h:7

c10::NUMABind
void NUMABind(int numa_node_id)
Bind to a given NUMA node.
Definition: numa.cpp:112

c10::GetNumNUMANodes
int GetNumNUMANodes()
Get number of NUMA nodes.
Definition: numa.cpp:119