Searched refs:pci_bus_id (Results 1 – 11 of 11) sorted by relevance
| /external/tensorflow/tensorflow/tools/test/ |
| D | gpu_info_lib.py | 152 pci_bus_id = " " * 13
153 rc = libcudart.cudaDeviceGetPCIBusId(ct.c_char_p(pci_bus_id), 13, i)
159 info.bus_id = pci_bus_id
|
| /external/libdrm/tests/amdgpu/ |
| D | amdgpu_test.c | 487 int pci_bus_id = -1; /* By default PC bus ID is not specified */ in main() local
514 pci_bus_id = atoi(optarg); in main()
554 if (pci_bus_id > 0 || pci_device_id) { in main()
556 test_device_index = amdgpu_find_device(pci_bus_id, in main()
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_gpu_executor.cc | 940 static int TryToReadNumaNode(const string &pci_bus_id, int device_ordinal) { in TryToReadNumaNode() argument
954 if (pci_bus_id.empty()) { in TryToReadNumaNode()
960 absl::StrFormat("/sys/bus/pci/devices/%s/numa_node", pci_bus_id); in TryToReadNumaNode()
1027 string pci_bus_id = GpuDriver::GetPCIBusID(device); in CreateDeviceDescription() local
1030 pci_bus_id = absl::AsciiStrToLower(pci_bus_id); in CreateDeviceDescription()
1031 builder.set_pci_bus_id(pci_bus_id); in CreateDeviceDescription()
1034 int numa_node = TryToReadNumaNode(pci_bus_id, device_ordinal); in CreateDeviceDescription()
|
| D | cuda_driver.cc | 1395 string pci_bus_id; in GetPCIBusID() local
1402 return pci_bus_id; in GetPCIBusID()
1404 pci_bus_id = chars.begin(); in GetPCIBusID()
1405 return pci_bus_id; in GetPCIBusID()
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_gpu_executor.cc | 840 static int TryToReadNumaNode(const string& pci_bus_id, int device_ordinal) { in TryToReadNumaNode() argument
872 string pci_bus_id = GpuDriver::GetPCIBusID(device); in CreateDeviceDescription() local
875 pci_bus_id = absl::AsciiStrToLower(pci_bus_id); in CreateDeviceDescription()
876 builder.set_pci_bus_id(pci_bus_id); in CreateDeviceDescription()
879 int numa_node = TryToReadNumaNode(pci_bus_id, device_ordinal); in CreateDeviceDescription()
|
| D | rocm_driver.cc | 1234 string pci_bus_id; in GetPCIBusID() local
1242 return pci_bus_id; in GetPCIBusID()
1244 pci_bus_id = chars.begin(); in GetPCIBusID()
1245 return pci_bus_id; in GetPCIBusID()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | gpu_utils.cc | 154 log.set_device_pci_bus_id(stream_exec->GetDeviceDescription().pci_bus_id()); in LogConvAutotuneResults()
200 log.set_device_pci_bus_id(stream_exec->GetDeviceDescription().pci_bus_id()); in LogFusedConvForwardAutotuneResults()
|
| /external/tensorflow/tensorflow/core/protobuf/ |
| D | autotuning.proto | 82 // stream_executor::DeviceDescription::pci_bus_id.
|
| /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 | 1190 ", pci bus id: ", desc.pci_bus_id(), in GetShortDeviceDescription()
1196 ", pci bus id: ", desc.pci_bus_id()); in GetShortDeviceDescription()
1372 << " numa: " << numa_node << " pci: " << desc->pci_bus_id() in GetDeviceLocalities()
1559 << "\npciBusID: " << description->pci_bus_id() in GetValidDeviceIds()
1577 << "\npciBusID: " << description->pci_bus_id() in GetValidDeviceIds()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_conv_algorithm_picker.cc | 583 stream_exec_->GetDeviceDescription().pci_bus_id()); in PickBestAlgorithmNoCacheCuda()
|