diff options
Diffstat (limited to 'tensorflow/core/common_runtime/device_set_test.cc')
-rw-r--r-- | tensorflow/core/common_runtime/device_set_test.cc | 9 |
1 files changed, 7 insertions, 2 deletions
diff --git a/tensorflow/core/common_runtime/device_set_test.cc b/tensorflow/core/common_runtime/device_set_test.cc index 21d3be2f61..550fbf568e 100644 --- a/tensorflow/core/common_runtime/device_set_test.cc +++ b/tensorflow/core/common_runtime/device_set_test.cc @@ -68,12 +68,17 @@ TEST_F(DeviceSetTest, PrioritizedDeviceTypeList) { (std::vector<DeviceType>{DeviceType(DEVICE_GPU), DeviceType(DEVICE_CPU)}), types()); + AddDevice("SYCL", "/job:a/replica:0/task:0/device:sycl:0"); + EXPECT_EQ( + (std::vector<DeviceType>{DeviceType(DEVICE_SYCL), DeviceType(DEVICE_GPU), + DeviceType(DEVICE_CPU)}), types()); + AddDevice("T1", "/job:a/replica:0/task:0/device:T1:0"); AddDevice("T1", "/job:a/replica:0/task:0/device:T1:1"); AddDevice("T2", "/job:a/replica:0/task:0/device:T2:0"); EXPECT_EQ( - (std::vector<DeviceType>{DeviceType("T1"), DeviceType("T2"), - DeviceType(DEVICE_GPU), DeviceType(DEVICE_CPU)}), + (std::vector<DeviceType>{DeviceType(DEVICE_SYCL), DeviceType("T1"), + DeviceType("T2"), DeviceType(DEVICE_GPU), DeviceType(DEVICE_CPU)}), types()); } |