experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Add ListPhysicalDevices to SYCLDeviceFactory

Browse Source 
PiperOrigin-RevId: 243620212
This commit is contained in:
Gaurav Jain 2019-04-15 08:29:44 -07:00 committed by TensorFlower Gardener
parent 221a5146df
commit e656cf4837
1 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
tensorflow/core/common_runtime/sycl/sycl_device_factory.cc
View File

@ -24,8 +24,12 @@ namespace tensorflow {
class SYCLDeviceFactory : public DeviceFactory { class SYCLDeviceFactory : public DeviceFactory {
public: public:
Status ListPhysicalDevices(std::vector<string>* devices) override {
return tensorflow::Status::OK();
}
Status CreateDevices(const SessionOptions& options, const string& name_prefix, Status CreateDevices(const SessionOptions& options, const string& name_prefix,
std::vector<Device *> *devices) override { std::vector<std::unique_ptr<Device>>* devices) override {
auto syclInterface = GSYCLInterface::instance(); auto syclInterface = GSYCLInterface::instance();
size_t n = 1; size_t n = 1;