device/oneapi/queue.h

Go to the documentation of this file.

/* SPDX-FileCopyrightText: 2021-2022 Intel Corporation
 *
 * SPDX-License-Identifier: Apache-2.0 */
 
#pragma once
 
#ifdef WITH_ONEAPI
 
#  include "device/kernel.h"
#  include "device/memory.h"
#  include "device/queue.h"
 
#  include "device/oneapi/device.h"
#  include "kernel/device/oneapi/kernel.h"
 
CCL_NAMESPACE_BEGIN
 
class OneapiDevice;
class device_memory;
 
/* Base class for OneAPI queues. */
class OneapiDeviceQueue : public DeviceQueue {
 public:
  explicit OneapiDeviceQueue(OneapiDevice *device);
  ~OneapiDeviceQueue();
 
  virtual int num_concurrent_states(const size_t state_size) const override;
 
  virtual int num_concurrent_busy_states(const size_t state_size) const override;
 
  virtual int num_sort_partition_elements() const override;
 
  virtual void init_execution() override;
 
  virtual bool enqueue(DeviceKernel kernel,
                       const int kernel_work_size,
                       DeviceKernelArguments const &args) override;
 
  virtual bool synchronize() override;
 
  virtual void zero_to_device(device_memory &mem) override;
  virtual void copy_to_device(device_memory &mem) override;
  virtual void copy_from_device(device_memory &mem) override;
 
  virtual bool supports_local_atomic_sort() const override
  {
    return true;
  }
 
 protected:
  OneapiDevice *oneapi_device_;
  KernelContext *kernel_context_;
};
 
CCL_NAMESPACE_END
 
#endif /* WITH_ONEAPI */