OSSForks/tensorflow
1
0
Fork 0
mirror of https://github.com/zebrajr/tensorflow.git synced 2025-12-07 00:20:20 +01:00
Code Issues Actions 5 Packages Projects Releases Wiki Activity

[OpenCL] Improves device reporting (#10462)

Browse Source 
Prints: id, type, name, vendor and profile of the device
This commit is contained in:
Luke Iwanski 2017-06-07 00:43:15 +01:00 committed by Benoit Steiner
parent bf4c3dd6ba
commit 9b2c1af631
1 changed files with 29 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

30
tensorflow/core/common_runtime/sycl/sycl_device.h
View File

@ -65,6 +65,11 @@ class GSYCLInterface
if(!found_device) {
// Currently Intel GPU is not supported
LOG(FATAL) << "No OpenCL GPU nor CPU found that is supported by ComputeCpp";
} else {
LOG(INFO) << "Found following OpenCL devices:";
for (int i = 0; i < device_list.size(); i++) {
LOG(INFO) << GetShortDeviceDescription(i);
}
}
}
@ -153,7 +158,30 @@ class GSYCLInterface
}
string GetShortDeviceDescription(int device_id = 0) {
return strings::StrCat("device: ", device_id, " ,name: SYCL");
auto _device = GetSYCLAllocator(device_id)
->getSyclDevice()
->sycl_queue()
.get_device();
auto _name = _device.get_info<cl::sycl::info::device::name>();
auto _vendor = _device.get_info<cl::sycl::info::device::vendor>();
auto _profile = _device.get_info<cl::sycl::info::device::profile>();
std::string _type;
if (_device.is_host()) {
_type = "Host";
} else if (_device.is_cpu()) {
_type = "CPU";
} else if (_device.is_gpu()) {
_type = "GPU";
} else if (_device.is_accelerator()) {
_type = "Accelerator";
} else {
_type = "Unknown";
}
return strings::StrCat("id: ", device_id, " ,type: ", _type, " ,name: ",
_name.c_str(), " ,vendor: ", _vendor.c_str(),
" ,profile: ", _profile.c_str());
}
};