2023-06-14 16:52:36 +10:00
|
|
|
/* SPDX-FileCopyrightText: 2021-2022 Intel Corporation
|
|
|
|
|
*
|
|
|
|
|
* SPDX-License-Identifier: Apache-2.0 */
|
2022-06-29 12:58:04 +02:00
|
|
|
|
|
|
|
|
#pragma once
|
|
|
|
|
|
|
|
|
|
#ifdef WITH_ONEAPI
|
|
|
|
|
|
|
|
|
|
# include "device/kernel.h"
|
|
|
|
|
# include "device/memory.h"
|
|
|
|
|
# include "device/queue.h"
|
|
|
|
|
|
|
|
|
|
# include "device/oneapi/device.h"
|
2022-10-06 18:35:51 +02:00
|
|
|
# include "kernel/device/oneapi/kernel.h"
|
2022-06-29 12:58:04 +02:00
|
|
|
|
|
|
|
|
CCL_NAMESPACE_BEGIN
|
|
|
|
|
|
|
|
|
|
class OneapiDevice;
|
|
|
|
|
class device_memory;
|
|
|
|
|
|
2022-06-30 12:14:22 +10:00
|
|
|
/* Base class for OneAPI queues. */
|
2022-06-29 12:58:04 +02:00
|
|
|
class OneapiDeviceQueue : public DeviceQueue {
|
|
|
|
|
public:
|
|
|
|
|
explicit OneapiDeviceQueue(OneapiDevice *device);
|
|
|
|
|
~OneapiDeviceQueue();
|
|
|
|
|
|
|
|
|
|
virtual int num_concurrent_states(const size_t state_size) const override;
|
|
|
|
|
|
2022-10-24 10:23:56 +01:00
|
|
|
virtual int num_concurrent_busy_states(const size_t state_size) const override;
|
2022-06-29 12:58:04 +02:00
|
|
|
|
2024-01-31 17:25:34 +01:00
|
|
|
virtual int num_sort_partition_elements() const override;
|
|
|
|
|
|
2022-06-29 12:58:04 +02:00
|
|
|
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;
|
|
|
|
|
|
2023-12-12 13:11:30 +01:00
|
|
|
virtual bool supports_local_atomic_sort() const override
|
2023-05-17 00:20:46 +02:00
|
|
|
{
|
|
|
|
|
return true;
|
|
|
|
|
}
|
|
|
|
|
|
2022-06-29 12:58:04 +02:00
|
|
|
protected:
|
|
|
|
|
OneapiDevice *oneapi_device_;
|
|
|
|
|
KernelContext *kernel_context_;
|
|
|
|
|
};
|
|
|
|
|
|
|
|
|
|
CCL_NAMESPACE_END
|
|
|
|
|
|
|
|
|
|
#endif /* WITH_ONEAPI */
|