| /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 | 539 int pci_bus_id = -1; /* By default PC bus ID is not specified */ in main() local
566 pci_bus_id = atoi(optarg); in main()
606 if (pci_bus_id > 0 || pci_device_id) { in main()
608 test_device_index = amdgpu_find_device(pci_bus_id, in main()
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_gpu_executor.cc | 899 static int TryToReadNumaNode(const std::string& pci_bus_id, in TryToReadNumaNode() argument
914 if (pci_bus_id.empty()) { in TryToReadNumaNode()
920 absl::StrFormat("/sys/bus/pci/devices/%s/numa_node", pci_bus_id); in TryToReadNumaNode()
988 std::string pci_bus_id = GpuDriver::GetPCIBusID(device); in CreateDeviceDescription() local
991 pci_bus_id = absl::AsciiStrToLower(pci_bus_id); in CreateDeviceDescription()
992 builder.set_pci_bus_id(pci_bus_id); in CreateDeviceDescription()
995 int numa_node = TryToReadNumaNode(pci_bus_id, device_ordinal); in CreateDeviceDescription()
|
| D | cuda_driver.cc | 1558 std::string pci_bus_id; in GetPCIBusID() local
1565 return pci_bus_id; in GetPCIBusID()
1567 pci_bus_id = chars.begin(); in GetPCIBusID()
1568 return pci_bus_id; in GetPCIBusID()
|
| /external/tensorflow/tensorflow/c/experimental/stream_executor/ |
| D | stream_executor_test.cc | 933 static const char* pci_bus_id = "TestPCIBusId"; in TEST_F() local
939 params->device->pci_bus_id = pci_bus_id; in TEST_F()
952 ASSERT_EQ(description.pci_bus_id(), "TestPCIBusId"); in TEST_F()
960 static const char* pci_bus_id = "TestPCIBusId"; in TEST_F() local
966 params->device->pci_bus_id = pci_bus_id; in TEST_F()
978 ASSERT_EQ(description.pci_bus_id(), "TestPCIBusId"); in TEST_F()
|
| D | stream_executor.h | 182 const char* pci_bus_id; member
185 #define SP_DEVICE_STRUCT_SIZE TF_OFFSET_OF_END(SP_Device, pci_bus_id)
|
| D | stream_executor.cc | 616 if (device_.pci_bus_id != nullptr) { in CreateDeviceDescription()
617 builder.set_pci_bus_id(device_.pci_bus_id); in CreateDeviceDescription()
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_gpu_executor.cc | 804 static int TryToReadNumaNode(const string& pci_bus_id, int device_ordinal) { in TryToReadNumaNode() argument
842 string pci_bus_id = GpuDriver::GetPCIBusID(device); in CreateDeviceDescription() local
845 pci_bus_id = absl::AsciiStrToLower(pci_bus_id); in CreateDeviceDescription()
846 builder.set_pci_bus_id(pci_bus_id); in CreateDeviceDescription()
849 int numa_node = TryToReadNumaNode(pci_bus_id, device_ordinal); in CreateDeviceDescription()
|
| D | rocm_driver.cc | 1256 string pci_bus_id; in GetPCIBusID() local
1264 return pci_bus_id; in GetPCIBusID()
1266 pci_bus_id = chars.begin(); in GetPCIBusID()
1267 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()
201 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/tpu/ |
| D | c_api_decl.h | 122 char* pci_bus_id; member
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | device_description.h | 60 const std::string &pci_bus_id() const { return pci_bus_id_; } in pci_bus_id() function
|
| /external/crosvm/x86_64/src/ |
| D | mptable.rs | 202 for pci_bus_id in 0..isa_bus_id { in setup_mptable()
206 busid: pci_bus_id, in setup_mptable()
|
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | gpu_device.cc | 1391 ", pci bus id: ", desc.pci_bus_id(), in GetShortDeviceDescription()
1397 ", pci bus id: ", desc.pci_bus_id()); in GetShortDeviceDescription()
1584 << " numa: " << numa_node << " pci: " << desc->pci_bus_id() in GetDeviceLocalities()
1771 << "\npciBusID: " << description->pci_bus_id() in GetValidDeviceIds()
1785 << "\npciBusID: " << description->pci_bus_id() in GetValidDeviceIds()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_conv_algorithm_picker.cc | 596 stream_exec_->GetDeviceDescription().pci_bus_id()); in PickBestAlgorithmNoCacheCuda()
|