| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | local_device.cc | 73 explicit EigenThreadPoolInfo(const SessionOptions& options, int numa_node, in EigenThreadPoolInfo()
85 if (numa_node != port::kNUMANoAffinity) { in EigenThreadPoolInfo()
94 thread_opts.numa_node = numa_node; in EigenThreadPoolInfo()
97 options.env, thread_opts, strings::StrCat("numa_", numa_node, "_Eigen"), in EigenThreadPoolInfo()
136 int numa_node = attributes.locality().numa_node(); in LocalDevice() local
138 DCHECK_LT(numa_node, num_numa_nodes); in LocalDevice()
140 ProcessState::singleton()->GetCPUAllocator(numa_node); in LocalDevice()
141 while (numa_node >= global_tp_info_.size()) { in LocalDevice()
144 if (!global_tp_info_[numa_node]) { in LocalDevice()
145 global_tp_info_[numa_node] = new LocalDevice::EigenThreadPoolInfo( in LocalDevice()
[all …]
|
| D | process_state.cc | 62 Allocator* ProcessState::GetCPUAllocator(int numa_node) { in GetCPUAllocator() argument
63 if (!numa_enabled_ || numa_node == port::kNUMANoAffinity) numa_node = 0; in GetCPUAllocator()
65 while (cpu_allocators_.size() <= static_cast<size_t>(numa_node)) { in GetCPUAllocator()
81 numa_enabled_ ? numa_node : port::kNUMANoAffinity, in GetCPUAllocator()
107 << " numa_node=" << numa_node; in GetCPUAllocator()
122 return cpu_allocators_[numa_node]; in GetCPUAllocator()
|
| D | threadpool_device_factory.cc | 44 int numa_node = i % num_numa_nodes; in CreateDevices() local
45 if (numa_node != i) { in CreateDevices()
49 << numa_node; in CreateDevices()
52 dev_locality.set_numa_node(numa_node); in CreateDevices()
55 ProcessState::singleton()->GetCPUAllocator(numa_node)); in CreateDevices()
|
| D | pool_allocator.h | 151 BasicCPUAllocator(int numa_node, const std::vector<Visitor>& alloc_visitors, in BasicCPUAllocator() argument
153 : SubAllocator(alloc_visitors, free_visitors), numa_node_(numa_node) {} in BasicCPUAllocator()
|
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | gpu_process_state.cc | 74 int numa_node = se->GetDeviceDescription().numa_node(); in BusIdForGPU() local
77 return numa_node >= 0 ? numa_node : 0; in BusIdForGPU()
180 Allocator* GPUProcessState::GetGpuHostAllocator(int numa_node) { in GetGpuHostAllocator() argument
184 return process_state_->GetCPUAllocator(numa_node); in GetGpuHostAllocator()
186 if (numa_node == port::kNUMANoAffinity) { in GetGpuHostAllocator()
187 numa_node = 0; in GetGpuHostAllocator()
201 if (static_cast<int>(gpu_host_allocators_.size()) > numa_node) { in GetGpuHostAllocator()
224 while (static_cast<int>(gpu_host_allocators_.size()) <= numa_node) { in GetGpuHostAllocator()
225 while (gpu_host_alloc_visitors_.size() <= numa_node) { in GetGpuHostAllocator()
228 while (gpu_host_free_visitors_.size() <= numa_node) { in GetGpuHostAllocator()
[all …]
|
| D | gpu_device_factory.cc | 86 numa_node_(locality.numa_node()) { in GPUCompatibleCPUDevice()
124 int numa_node = i % num_numa_nodes; in CreateDevices() local
126 locality.set_numa_node(numa_node); in CreateDevices()
129 ProcessState::singleton()->GetCPUAllocator(numa_node))); in CreateDevices()
|
| D | gpu_process_state.h | 86 virtual Allocator* GetGpuHostAllocator(int numa_node);
102 virtual void AddGpuHostAllocVisitor(int numa_node,
107 virtual void AddGpuHostFreeVisitor(int numa_node,
|
| D | gpu_host_allocator.h | 29 explicit GpuHostAllocator(se::StreamExecutor* stream_exec, int numa_node, in GpuHostAllocator() argument
34 numa_node_(numa_node) { in GpuHostAllocator()
|
| D | gpu_device.cc | 699 int numa_node = attributes().locality().numa_node(); in MakeTensorFromProto() local
700 Tensor copy(cpu_allocator(numa_node), DT_VARIANT, parsed.shape()); in MakeTensorFromProto()
1193 int numa_node = dev_locality.numa_node(); in CreateGPUDevice() local
1221 gpu_allocator, ProcessState::singleton()->GetCPUAllocator(numa_node)); in CreateGPUDevice()
1293 int numa_node = desc.numa_node(); in GetDeviceLocalities() local
1294 if (numa_node < 0) { in GetDeviceLocalities()
1305 numa_node = 0; in GetDeviceLocalities()
1308 dev_locality.set_numa_node(numa_node); in GetDeviceLocalities()
1309 dev_locality.set_bus_id(numa_node + 1); in GetDeviceLocalities()
1346 << " numa: " << numa_node << " pci: " << desc.pci_bus_id() in GetDeviceLocalities()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | allocator_registry.cc | 90 SubAllocator* AllocatorFactoryRegistry::GetSubAllocator(int numa_node) { in GetSubAllocator() argument
112 if (numa_node != port::kNUMANoAffinity) { in GetSubAllocator()
113 CHECK_LE(numa_node, port::NUMANumNodes()); in GetSubAllocator()
114 index = 1 + numa_node; in GetSubAllocator()
121 best_entry->factory->CreateSubAllocator(numa_node)); in GetSubAllocator()
|
| D | allocator_registry.h | 44 virtual SubAllocator* CreateSubAllocator(int numa_node) = 0;
51 virtual Allocator* GetCPUAllocator(int numa_node) = 0;
78 SubAllocator* GetSubAllocator(int numa_node);
|
| D | allocator.cc | 185 SubAllocator* CreateSubAllocator(int numa_node) override { in CreateSubAllocator() argument
224 Allocator* cpu_allocator(int numa_node) { in cpu_allocator() argument
232 return ps->GetCPUAllocator(numa_node); in cpu_allocator()
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_platform.cc | 56 min_numa_node_ = exec->GetDeviceDescription().numa_node(); in InspectNumaNodes()
60 std::min(min_numa_node_, exec->GetDeviceDescription().numa_node()); in InspectNumaNodes()
62 exec->GetDeviceDescription().numa_node() + 1); in InspectNumaNodes()
77 return exec->GetDeviceDescription().numa_node() - min_numa_node_; in DeviceToBus()
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_platform.cc | 91 min_numa_node_ = exec->GetDeviceDescription().numa_node(); in InspectNumaNodes()
95 std::min(min_numa_node_, exec->GetDeviceDescription().numa_node()); in InspectNumaNodes()
97 exec->GetDeviceDescription().numa_node() + 1); in InspectNumaNodes()
112 return exec->GetDeviceDescription().numa_node() - min_numa_node_; in DeviceToBus()
|
| /external/strace/xlat/ |
| D | numa_node.h | 18 # error static const struct xlat numa_node in mpers mode
23 const struct xlat numa_node[] = {
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/data/ |
| D | make_numa_aware.cc | 52 auto* numa_node = graph.AddNode(MakeNumaAwareNode(node, &graph)); in OptimizeAndCollectStats() local
53 TF_RETURN_IF_ERROR(graph.UpdateFanouts(node.name(), numa_node->name())); in OptimizeAndCollectStats()
|
| /external/strace/tests-mx32/ |
| D | bpf.c | 285 .numa_node = 3141592653,
307 .numa_node = -1,
328 .numa_node = -1,
346 .numa_node = -1,
365 .numa_node = -1,
367 .size = offsetofend(struct BPF_MAP_CREATE_struct, numa_node),
385 .numa_node = -1,
407 .numa_node = -1,
|
| /external/strace/tests-m32/ |
| D | bpf.c | 285 .numa_node = 3141592653,
307 .numa_node = -1,
328 .numa_node = -1,
346 .numa_node = -1,
365 .numa_node = -1,
367 .size = offsetofend(struct BPF_MAP_CREATE_struct, numa_node),
385 .numa_node = -1,
407 .numa_node = -1,
|
| /external/strace/tests/ |
| D | bpf.c | 285 .numa_node = 3141592653,
307 .numa_node = -1,
328 .numa_node = -1,
346 .numa_node = -1,
365 .numa_node = -1,
367 .size = offsetofend(struct BPF_MAP_CREATE_struct, numa_node),
385 .numa_node = -1,
407 .numa_node = -1,
|
| /external/tensorflow/tensorflow/c/ |
| D | env.cc | 166 options->numa_node = -1; in TF_DefaultThreadOptions()
175 cc_options.numa_node = options->numa_node; in TF_StartThread()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | device_resolver_distributed_test.cc | 46 int numa_node) { in NewDevice() argument
56 attr.mutable_locality()->set_numa_node(numa_node); in NewDevice()
201 EXPECT_EQ(parsed.id, locality.numa_node()); in TEST_F()
|
| /external/tensorflow/tensorflow/core/platform/posix/ |
| D | port.cc | 241 hwloc_obj_t numa_node = GetHWLocTypeIndex(HWLOC_OBJ_NUMANODE, node); in NUMAMalloc() local
242 if (numa_node) { in NUMAMalloc()
244 numa_node->nodeset, HWLOC_MEMBIND_BIND, in NUMAMalloc()
|
| /external/tensorflow/tensorflow/contrib/verbs/ |
| D | rdma_mgr.cc | 266 SubAllocator::Visitor alloc_visitor = [](void* ptr, int numa_node, in RegMemVisitors()
269 ptr, num_bytes, strings::StrCat("CPU:", numa_node)); in RegMemVisitors()
271 SubAllocator::Visitor free_visitor = [](void* ptr, int numa_node, in RegMemVisitors() argument
|
| /external/tensorflow/tensorflow/contrib/gdr/ |
| D | gdr_memory_manager.cc | 236 SubAllocator::Visitor alloc_visitor = [this](void* ptr, int numa_node, in Init()
239 << numa_node; in Init()
240 InsertMemoryRegion(ptr, num_bytes, strings::StrCat("CPU:", numa_node)); in Init()
242 SubAllocator::Visitor free_visitor = [this](void* ptr, int numa_node, in Init() argument
245 << numa_node; in Init()
|
| /external/tensorflow/tensorflow/core/lib/core/ |
| D | threadpool.cc | 58 if (thread_options_.numa_node != port::kNUMANoAffinity) { in CreateThread()
59 port::NUMASetThreadNodeAffinity(thread_options_.numa_node); in CreateThread()
|