Home
last modified time | relevance | path

Searched defs:numa_node (Results 1 – 25 of 64) sorted by relevance

123

/external/tensorflow/tensorflow/core/common_runtime/gpu/
Dgpu_process_state.cc91 int numa_node = se->GetDeviceDescription().numa_node(); in BusIdForGPU() local
291 Allocator* GPUProcessState::GetGpuHostAllocator(int numa_node) { in GetGpuHostAllocator()
409 int numa_node, const SubAllocator::Visitor& visitor) { in AddGpuHostAllocVisitor()
424 int numa_node, const SubAllocator::Visitor& visitor) { in AddGpuHostFreeVisitor()
Dpool_allocator_test.cc135 [&alloc_count, &alloc_size](void* ptr, int numa_node, int64_t size) { in TEST()
142 [&free_count, &free_size](void* ptr, int numa_node, int64_t size) { in TEST()
Dgpu_device_factory.cc131 int numa_node = i % num_numa_nodes; in CreateDevices() local
/external/tensorflow/tensorflow/core/common_runtime/pluggable_device/
Dpluggable_device_process_state.cc74 int numa_node = se->GetDeviceDescription().numa_node(); in BusIdForPluggableDevice() local
141 int numa_node) { in GetPluggableDeviceHostAllocator()
Dpluggable_device_factory.cc240 int numa_node = dev_locality.numa_node(); in CreatePluggableDevice() local
298 int numa_node = desc->numa_node(); in GetDeviceLocalities() local
/external/python/cpython3/Objects/mimalloc/
Darena.c50 int numa_node; // associated NUMA node member
273 static void* mi_arena_try_alloc_at_id(mi_arena_id_t arena_id, bool match_numa_node, int numa_node, … in mi_arena_try_alloc_at_id()
302 static mi_decl_noinline void* mi_arena_try_alloc(int numa_node, size_t size, size_t alignment, in mi_arena_try_alloc()
373 const int numa_node = _mi_os_numa_node(tld); // current numa node in _mi_arena_alloc_aligned() local
749 static bool mi_manage_os_memory_ex2(void* start, size_t size, bool is_large, int numa_node, bool ex… in mi_manage_os_memory_ex2()
800 …* start, size_t size, bool is_committed, bool is_large, bool is_zero, int numa_node, bool exclusiv… in mi_manage_os_memory_ex()
827 …* start, size_t size, bool is_committed, bool is_large, bool is_zero, int numa_node) mi_attr_noexc… in mi_manage_os_memory()
874 int mi_reserve_huge_os_pages_at_ex(size_t pages, int numa_node, size_t timeout_msecs, bool exclusiv… in mi_reserve_huge_os_pages_at_ex()
896 int mi_reserve_huge_os_pages_at(size_t pages, int numa_node, size_t timeout_msecs) mi_attr_noexcept… in mi_reserve_huge_os_pages_at()
912 for (size_t numa_node = 0; numa_node < numa_count && pages > 0; numa_node++) { in mi_reserve_huge_os_pages_interleave() local
/external/tensorflow/tensorflow/c/
Denv.h49 int numa_node; member
/external/tensorflow/tensorflow/core/common_runtime/
Dthreadpool_device_factory.cc50 int numa_node = i % num_numa_nodes; in CreateDevices() local
Dprocess_state.cc64 Allocator* ProcessState::GetCPUAllocator(int numa_node) { in GetCPUAllocator()
Dpool_allocator.h156 BasicCPUAllocator(int numa_node, const std::vector<Visitor>& alloc_visitors, in BasicCPUAllocator()
Dlocal_device.cc129 int numa_node = attributes.locality().numa_node(); in LocalDevice() local
Dthreadpool_device.cc262 virtual SubAllocator* CreateSubAllocator(int numa_node) { in CreateSubAllocator()
/external/tensorflow/tensorflow/core/common_runtime/device/
Ddevice_host_allocator.h29 explicit DeviceHostAllocator(se::StreamExecutor* stream_exec, int numa_node, in DeviceHostAllocator()
/external/tensorflow/tensorflow/core/framework/
Ddevice_attributes.proto27 int32 numa_node = 2; field
Dallocator_registry.cc90 SubAllocator* AllocatorFactoryRegistry::GetSubAllocator(int numa_node) { in GetSubAllocator()
Dallocator.cc79 Allocator* cpu_allocator(int numa_node) { in cpu_allocator()
Dcpu_allocator_impl.cc186 SubAllocator* CreateSubAllocator(int numa_node) override { in CreateSubAllocator()
/external/pytorch/c10/util/
Dnuma.cpp49 int numa_node = -1; in GetNUMANode() local
/external/v4l-utils/utils/keytable/
Dbpf_load.h15 unsigned int numa_node; member
Dbpf_load.c91 static int build_raw_map(struct bpf_map_data *map, struct raw_entry *raw, int numa_node) in build_raw_map()
169 int i, numa_node; in load_maps() local
/external/tensorflow/tensorflow/tsl/platform/default/
Dport.cc119 int MaxParallelism(int numa_node) { in MaxParallelism()
287 hwloc_obj_t numa_node = GetHWLocTypeIndex(HWLOC_OBJ_NUMANODE, node); in NUMAMalloc() local
/external/python/cpython3/Objects/mimalloc/prim/windows/
Dprim.c307 static void* _mi_prim_alloc_huge_os_pagesx(void* hint_addr, size_t size, int numa_node) in _mi_prim_alloc_huge_os_pagesx()
348 int _mi_prim_alloc_huge_os_pages(void* hint_addr, size_t size, int numa_node, bool* is_zero, void**… in _mi_prim_alloc_huge_os_pages()
360 USHORT numa_node = 0; in _mi_prim_numa_node() local
/external/tensorflow/tensorflow/core/kernels/
Dconstant_op.cc253 int numa_node = ctx->device()->NumaNode(); in Compute() local
/external/python/cpython3/Objects/mimalloc/prim/wasi/
Dprim.c159 int _mi_prim_alloc_huge_os_pages(void* hint_addr, size_t size, int numa_node, bool* is_zero, void**… in _mi_prim_alloc_huge_os_pages()
/external/tensorflow/tensorflow/tsl/platform/windows/
Dport.cc70 int MaxParallelism(int numa_node) { in MaxParallelism()

123