Searched refs:pci_bus_id (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/tools/test/ |
| D | gpu_info_lib.py | 147 pci_bus_id = " " * 13
148 rc = libcudart.cudaDeviceGetPCIBusId(ct.c_char_p(pci_bus_id), 13, i)
154 info.bus_id = pci_bus_id
|
| /external/libdrm/tests/amdgpu/ |
| D | amdgpu_test.c | 450 int pci_bus_id = -1; /* By default PC bus ID is not specified */ in main() local
477 pci_bus_id = atoi(optarg); in main()
517 if (pci_bus_id > 0 || pci_device_id) { in main()
519 test_device_index = amdgpu_find_device(pci_bus_id, in main()
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_gpu_executor.cc | 968 static int TryToReadNumaNode(const string &pci_bus_id, int device_ordinal) { in TryToReadNumaNode() argument
982 if (pci_bus_id.empty()) { in TryToReadNumaNode()
988 port::Printf("/sys/bus/pci/devices/%s/numa_node", pci_bus_id.c_str()); in TryToReadNumaNode()
1042 string pci_bus_id = GpuDriver::GetPCIBusID(device_); in PopulateDeviceDescription() local
1045 pci_bus_id = port::Lowercase(pci_bus_id); in PopulateDeviceDescription()
1046 builder.set_pci_bus_id(pci_bus_id); in PopulateDeviceDescription()
1049 int numa_node = TryToReadNumaNode(pci_bus_id, device_ordinal_); in PopulateDeviceDescription()
|
| D | cuda_driver.cc | 1551 string pci_bus_id; in GetPCIBusID() local
1558 return pci_bus_id; in GetPCIBusID()
1560 pci_bus_id = chars.begin(); in GetPCIBusID()
1561 return pci_bus_id; in GetPCIBusID()
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_gpu_executor.cc | 870 static int TryToReadNumaNode(const string& pci_bus_id, int device_ordinal) { in TryToReadNumaNode() argument
889 string pci_bus_id = GpuDriver::GetPCIBusID(device_); in PopulateDeviceDescription() local
892 pci_bus_id = port::Lowercase(pci_bus_id); in PopulateDeviceDescription()
893 builder.set_pci_bus_id(pci_bus_id); in PopulateDeviceDescription()
896 int numa_node = TryToReadNumaNode(pci_bus_id, device_ordinal_); in PopulateDeviceDescription()
|
| D | rocm_driver.cc | 1304 string pci_bus_id; in GetPCIBusID() local
1312 return pci_bus_id; in GetPCIBusID()
1314 pci_bus_id = chars.begin(); in GetPCIBusID()
1315 return pci_bus_id; in GetPCIBusID()
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | device_description.h | 60 const string &pci_bus_id() const { return pci_bus_id_; } in pci_bus_id() function
|
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | gpu_device.cc | 1172 desc.name(), ", pci bus id: ", desc.pci_bus_id(), in GetShortDeviceDescription()
1178 ", pci bus id: ", desc.pci_bus_id()); in GetShortDeviceDescription()
1346 << " numa: " << numa_node << " pci: " << desc.pci_bus_id() in GetDeviceLocalities()
1555 << "\npciBusID: " << description.pci_bus_id() << "\ntotalMemory: " in GetValidDeviceIds()
1568 << description.pci_bus_id() << "\nTotal memory: " in GetValidDeviceIds()
|