eccc9d8eba
Noticed while looking into oneAPI patch. Seems to be unused, without clear indication why/when it might be needed. Removing the function simplifies adding the new backend. Differential Revision: https://developer.blender.org/D14652
57 lines
1.3 KiB
C++
57 lines
1.3 KiB
C++
/* SPDX-License-Identifier: Apache-2.0
|
|
* Copyright 2011-2022 Blender Foundation */
|
|
|
|
#pragma once
|
|
|
|
#ifdef WITH_CUDA
|
|
|
|
# include "device/kernel.h"
|
|
# include "device/memory.h"
|
|
# include "device/queue.h"
|
|
|
|
# include "device/cuda/util.h"
|
|
|
|
CCL_NAMESPACE_BEGIN
|
|
|
|
class CUDADevice;
|
|
class device_memory;
|
|
|
|
/* Base class for CUDA queues. */
|
|
class CUDADeviceQueue : public DeviceQueue {
|
|
public:
|
|
CUDADeviceQueue(CUDADevice *device);
|
|
~CUDADeviceQueue();
|
|
|
|
virtual int num_concurrent_states(const size_t state_size) const override;
|
|
virtual int num_concurrent_busy_states() const override;
|
|
|
|
virtual void init_execution() override;
|
|
|
|
virtual bool enqueue(DeviceKernel kernel,
|
|
const int 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 CUstream stream()
|
|
{
|
|
return cuda_stream_;
|
|
}
|
|
|
|
virtual unique_ptr<DeviceGraphicsInterop> graphics_interop_create() override;
|
|
|
|
protected:
|
|
CUDADevice *cuda_device_;
|
|
CUstream cuda_stream_;
|
|
|
|
void assert_success(CUresult result, const char *operation);
|
|
};
|
|
|
|
CCL_NAMESPACE_END
|
|
|
|
#endif /* WITH_CUDA */
|