| 123456789101112131415161718192021222324252627282930313233343536373839 |
- #pragma once
- #include <c10/util/Logging.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
|
