Home
last modified time | relevance | path

Searched refs:platform_gpu_id (Results 1 – 20 of 20) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/gpu/
Dgpu_debug_allocator_test.cc37 const PlatformGpuId platform_gpu_id(0); in TEST() local
39 GpuIdUtil::ExecutorForPlatformGpuId(platform_gpu_id).ValueOrDie(), in TEST()
40 platform_gpu_id, false /*use_unified_memory*/, {}, {}); in TEST()
42 platform_gpu_id); in TEST()
44 GpuIdUtil::ExecutorForPlatformGpuId(platform_gpu_id).ValueOrDie(); in TEST()
65 const PlatformGpuId platform_gpu_id(0); in TEST()
67 GpuIdUtil::ExecutorForPlatformGpuId(platform_gpu_id).ValueOrDie(), in TEST()
68 platform_gpu_id, false /*use_unified_memory*/, {}, {}); in TEST()
70 platform_gpu_id); in TEST()
72 GpuIdUtil::ExecutorForPlatformGpuId(platform_gpu_id).ValueOrDie(); in TEST()
[all …]
Dgpu_bfc_allocator_test.cc53 PlatformGpuId platform_gpu_id(0); in TEST() local
55 GpuIdUtil::ExecutorForPlatformGpuId(platform_gpu_id).ValueOrDie(), in TEST()
56 platform_gpu_id, false /*use_unified_memory*/, {}, {}); in TEST()
86 PlatformGpuId platform_gpu_id(0); in TEST() local
88 GpuIdUtil::ExecutorForPlatformGpuId(platform_gpu_id).ValueOrDie(), in TEST()
89 platform_gpu_id, false /*use_unified_memory*/, {}, {}); in TEST()
148 PlatformGpuId platform_gpu_id(0); in TEST() local
150 GpuIdUtil::ExecutorForPlatformGpuId(platform_gpu_id).ValueOrDie(), in TEST()
151 platform_gpu_id, false /*use_unified_memory*/, {}, {}); in TEST()
187 PlatformGpuId platform_gpu_id(0); in TEST() local
[all …]
Dgpu_id_utils.h34 se::Platform* gpu_manager, PlatformGpuId platform_gpu_id) { in ExecutorForPlatformGpuId() argument
35 return gpu_manager->ExecutorForDevice(platform_gpu_id.value()); in ExecutorForPlatformGpuId()
38 PlatformGpuId platform_gpu_id) { in ExecutorForPlatformGpuId() argument
39 return ExecutorForPlatformGpuId(GPUMachineManager(), platform_gpu_id); in ExecutorForPlatformGpuId()
43 PlatformGpuId platform_gpu_id; in ExecutorForTfGpuId() local
45 GpuIdManager::TfToPlatformGpuId(tf_gpu_id, &platform_gpu_id)); in ExecutorForTfGpuId()
46 return ExecutorForPlatformGpuId(platform_gpu_id); in ExecutorForTfGpuId()
51 PlatformGpuId platform_gpu_id; in CheckValidTfGpuId() local
52 TF_CHECK_OK(GpuIdManager::TfToPlatformGpuId(tf_gpu_id, &platform_gpu_id)); in CheckValidTfGpuId()
54 CHECK_LT(platform_gpu_id.value(), visible_device_count) in CheckValidTfGpuId()
[all …]
Dgpu_id_manager.cc37 Status Insert(TfGpuId tf_gpu_id, PlatformGpuId platform_gpu_id) in Insert() argument
42 result = id_map_.insert({tf_gpu_id.value(), platform_gpu_id.value()}); in Insert()
44 if (!result.second && platform_gpu_id.value() != result.first->second) { in Insert()
49 platform_gpu_id.value(), " now, and ", result.first->second, in Insert()
60 bool Find(TfGpuId tf_gpu_id, PlatformGpuId* platform_gpu_id) const in Find()
65 *platform_gpu_id = result->second; in Find()
87 PlatformGpuId platform_gpu_id) { in InsertTfPlatformGpuIdPair() argument
88 return TfToPlatformGpuIdMap::singleton()->Insert(tf_gpu_id, platform_gpu_id); in InsertTfPlatformGpuIdPair()
92 PlatformGpuId* platform_gpu_id) { in TfToPlatformGpuId() argument
93 if (TfToPlatformGpuIdMap::singleton()->Find(tf_gpu_id, platform_gpu_id)) { in TfToPlatformGpuId()
Dgpu_device.cc133 PlatformGpuId platform_gpu_id; in Reinitialize() local
134 TF_CHECK_OK(GpuIdManager::TfToPlatformGpuId(tf_gpu_id, &platform_gpu_id)); in Reinitialize()
135 device_prop_ = &Eigen::m_deviceProperties[platform_gpu_id.value()]; in Reinitialize()
398 PlatformGpuId platform_gpu_id; in Init() local
400 GpuIdManager::TfToPlatformGpuId(tf_gpu_id_, &platform_gpu_id)); in Init()
401 gpu_device_info_->gpu_id = platform_gpu_id.value(); in Init()
785 int32 platform_gpu_id; in ParseVisibleDeviceList() local
786 if (!strings::safe_strto32(platform_gpu_id_str, &platform_gpu_id)) { in ParseVisibleDeviceList()
792 if (platform_gpu_id < 0 || in ParseVisibleDeviceList()
793 platform_gpu_id >= gpu_manager->VisibleDeviceCount()) { in ParseVisibleDeviceList()
[all …]
Dgpu_process_state.cc101 PlatformGpuId platform_gpu_id; in GetGPUAllocator() local
102 TF_CHECK_OK(GpuIdManager::TfToPlatformGpuId(tf_gpu_id, &platform_gpu_id)); in GetGPUAllocator()
109 GpuIdUtil::ExecutorForPlatformGpuId(platform_gpu_id).ValueOrDie(), in GetGPUAllocator()
110 platform_gpu_id, in GetGPUAllocator()
127 gpu_allocator = new GPUDebugAllocator(gpu_allocator, platform_gpu_id); in GetGPUAllocator()
128 gpu_allocator = new GPUNanResetAllocator(gpu_allocator, platform_gpu_id); in GetGPUAllocator()
134 new GPUcudaMallocAllocator(gpu_allocator, platform_gpu_id); in GetGPUAllocator()
141 md.dev_index = platform_gpu_id.value(); in GetGPUAllocator()
Dgpu_id_manager_test.cc26 PlatformGpuId platform_gpu_id; in TfToPlatformGpuId() local
27 TF_CHECK_OK(GpuIdManager::TfToPlatformGpuId(tf, &platform_gpu_id)); in TfToPlatformGpuId()
28 return platform_gpu_id; in TfToPlatformGpuId()
Dgpu_id_manager.h30 PlatformGpuId platform_gpu_id);
34 PlatformGpuId* platform_gpu_id);
Dgpu_debug_allocator.cc79 PlatformGpuId platform_gpu_id) in GPUDebugAllocator() argument
82 GpuIdUtil::ExecutorForPlatformGpuId(platform_gpu_id).ValueOrDie(); in GPUDebugAllocator()
157 PlatformGpuId platform_gpu_id) in GPUNanResetAllocator() argument
160 GpuIdUtil::ExecutorForPlatformGpuId(platform_gpu_id).ValueOrDie(); in GPUNanResetAllocator()
Dgpu_cudamalloc_allocator.cc31 PlatformGpuId platform_gpu_id) in GPUcudaMallocAllocator() argument
34 GpuIdUtil::ExecutorForPlatformGpuId(platform_gpu_id).ValueOrDie(); in GPUcudaMallocAllocator()
Dgpu_debug_allocator.h37 PlatformGpuId platform_gpu_id);
67 PlatformGpuId platform_gpu_id);
Dgpu_device.h103 PlatformGpuId platform_gpu_id; in gpu_id() local
104 TF_CHECK_OK(GpuIdManager::TfToPlatformGpuId(tf_gpu_id_, &platform_gpu_id)); in gpu_id()
105 return platform_gpu_id.value(); in gpu_id()
Dgpu_cudamalloc_allocator.h35 PlatformGpuId platform_gpu_id);
/external/tensorflow/tensorflow/core/grappler/clusters/
Dutils.cc77 DeviceProperties GetLocalGPUInfo(PlatformGpuId platform_gpu_id) { in GetLocalGPUInfo() argument
84 cudaGetDeviceProperties(&properties, platform_gpu_id.value()); in GetLocalGPUInfo()
120 hipGetDeviceProperties(&properties, platform_gpu_id.value()); in GetLocalGPUInfo()
160 PlatformGpuId platform_gpu_id; in GetDeviceInfo() local
161 Status s = GpuIdManager::TfToPlatformGpuId(tf_gpu_id, &platform_gpu_id); in GetDeviceInfo()
166 return GetLocalGPUInfo(platform_gpu_id); in GetDeviceInfo()
Dutils.h31 DeviceProperties GetLocalGPUInfo(PlatformGpuId platform_gpu_id);
Dsingle_machine.cc96 PlatformGpuId platform_gpu_id; in Provision() local
97 Status s = GpuIdManager::TfToPlatformGpuId(tf_gpu_id, &platform_gpu_id); in Provision()
102 attr = GetLocalGPUInfo(platform_gpu_id); in Provision()
/external/tensorflow/tensorflow/compiler/jit/
Dxla_gpu_device.cc44 int32 platform_gpu_id; in ParseVisibleDeviceList() local
45 if (!absl::SimpleAtoi(platform_gpu_id_str, &platform_gpu_id)) { in ParseVisibleDeviceList()
51 gpu_ids.insert(platform_gpu_id); in ParseVisibleDeviceList()
/external/tensorflow/tensorflow/compiler/tf2tensorrt/kernels/
Dtrt_engine_op.cc686 const int platform_gpu_id = in AllocateCalibrationResources() local
688 if (platform_gpu_id < 0) { in AllocateCalibrationResources()
695 platform_gpu_id, workspace_size_bytes]() { in AllocateCalibrationResources()
696 LOG(INFO) << "Starting calibration thread on device " << platform_gpu_id in AllocateCalibrationResources()
698 auto err = cudaSetDevice(platform_gpu_id); in AllocateCalibrationResources()
701 LOG(ERROR) << "Couldn't set cuda device to " << platform_gpu_id in AllocateCalibrationResources()
/external/tensorflow/tensorflow/core/grappler/costs/
Dutils.cc244 PlatformGpuId platform_gpu_id; in GetDeviceInfo() local
245 Status s = GpuIdManager::TfToPlatformGpuId(tf_gpu_id, &platform_gpu_id); in GetDeviceInfo()
248 platform_gpu_id = PlatformGpuId(parsed.id); in GetDeviceInfo()
250 return GetLocalGPUInfo(platform_gpu_id); in GetDeviceInfo()
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_graph.cc739 PlatformGpuId platform_gpu_id; in GetDeviceAndAllocator() local
740 Status s = GpuIdManager::TfToPlatformGpuId(tf_gpu_id, &platform_gpu_id); in GetDeviceAndAllocator()
743 << platform_gpu_id.value(); in GetDeviceAndAllocator()
744 cuda_device_id = platform_gpu_id.value(); in GetDeviceAndAllocator()