Home
last modified time | relevance | path

Searched refs:sycl_device_ (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/sycl/
Dsycl_allocator.cc23 : sycl_device_(new Eigen::SyclDevice(queue)) { in SYCLAllocator()
24 cl::sycl::queue& sycl_queue = sycl_device_->sycl_queue(); in SYCLAllocator()
31 if (sycl_device_) { in ~SYCLAllocator()
32 delete sycl_device_; in ~SYCLAllocator()
40 assert(sycl_device_); in AllocateRaw()
45 auto p = sycl_device_->allocate(num_bytes); in AllocateRaw()
46 const auto& allocated_buffer = sycl_device_->get_sycl_buffer(p); in AllocateRaw()
61 if (sycl_device_) { in DeallocateRaw()
62 const auto& buffer_to_delete = sycl_device_->get_sycl_buffer(ptr); in DeallocateRaw()
65 sycl_device_->deallocate(ptr); in DeallocateRaw()
[all …]
Dsycl_allocator.h41 if (sycl_device_) { in Synchronize()
42 sycl_device_->synchronize(); in Synchronize()
45 bool Ok() { return sycl_device_ && sycl_device_->ok(); } in Ok()
55 Eigen::SyclDevice* getSyclDevice() { return sycl_device_; } in getSyclDevice()
59 if (sycl_device_) { in ClearSYCLDevice()
60 delete sycl_device_; in ClearSYCLDevice() local
61 sycl_device_ = nullptr; in ClearSYCLDevice()
67 Eigen::SyclDevice* sycl_device_ GUARDED_BY(mu_); // owned