diff options
Diffstat (limited to 'tensorflow/core/common_runtime/sycl/sycl_device.cc')
-rw-r--r-- | tensorflow/core/common_runtime/sycl/sycl_device.cc | 26 |
1 files changed, 8 insertions, 18 deletions
diff --git a/tensorflow/core/common_runtime/sycl/sycl_device.cc b/tensorflow/core/common_runtime/sycl/sycl_device.cc index 17f5edd572..6e1a45b3ef 100644 --- a/tensorflow/core/common_runtime/sycl/sycl_device.cc +++ b/tensorflow/core/common_runtime/sycl/sycl_device.cc @@ -22,20 +22,10 @@ limitations under the License. #include "tensorflow/core/platform/tracing.h" namespace tensorflow { -std::mutex GSYCLInterface::mutex_; -GSYCLInterface *GSYCLInterface::s_instance = 0; - -void ShutdownSycl() { - GSYCLInterface::Reset(); -} - -void SYCLDevice::RegisterDevice() { - atexit(ShutdownSycl); -} SYCLDevice::~SYCLDevice() {} -void SYCLDevice::Compute(OpKernel *op_kernel, OpKernelContext *context) { +void SYCLDevice::Compute(OpKernel* op_kernel, OpKernelContext* context) { assert(context); if (port::Tracing::IsActive()) { // TODO(pbar) We really need a useful identifier of the graph node. @@ -46,16 +36,16 @@ void SYCLDevice::Compute(OpKernel *op_kernel, OpKernelContext *context) { op_kernel->Compute(context); } -Allocator *SYCLDevice::GetAllocator(AllocatorAttributes attr) { +Allocator* SYCLDevice::GetAllocator(AllocatorAttributes attr) { if (attr.on_host()) return cpu_allocator_; else return sycl_allocator_; } -Status SYCLDevice::MakeTensorFromProto(const TensorProto &tensor_proto, +Status SYCLDevice::MakeTensorFromProto(const TensorProto& tensor_proto, const AllocatorAttributes alloc_attrs, - Tensor *tensor) { + Tensor* tensor) { AllocatorAttributes attr; attr.set_on_host(true); Allocator* host_alloc = GetAllocator(attr); @@ -79,18 +69,18 @@ Status SYCLDevice::MakeTensorFromProto(const TensorProto &tensor_proto, } device_context_->CopyCPUTensorToDevice( - &parsed, this, ©, [&status](const Status &s) { status = s; }); + &parsed, this, ©, [&status](const Status& s) { status = s; }); *tensor = copy; } return status; } -Status SYCLDevice::FillContextMap(const Graph *graph, - DeviceContextMap *device_context_map) { +Status SYCLDevice::FillContextMap(const Graph* graph, + DeviceContextMap* device_context_map) { // Fill in the context map. It is OK for this map to contain // duplicate DeviceContexts so long as we increment the refcount. device_context_map->resize(graph->num_node_ids()); - for (Node *n : graph->nodes()) { + for (Node* n : graph->nodes()) { device_context_->Ref(); (*device_context_map)[n->id()] = device_context_; } |