d0c2e68e5f
* Use .empty() and .data() * Use nullptr instead of 0 * No else after return * Simple class member initialization * Add override for virtual methods * Include C++ instead of C headers * Remove some unused includes * Use default constructors * Always use braces * Consistent names in definition and declaration * Change typedef to using Pull Request: https://projects.blender.org/blender/blender/pulls/132361
57 lines
1.2 KiB
C++
57 lines
1.2 KiB
C++
/* SPDX-FileCopyrightText: 2011-2022 Blender Foundation
|
|
*
|
|
* SPDX-License-Identifier: Apache-2.0 */
|
|
|
|
#pragma once
|
|
|
|
#ifdef WITH_HIP
|
|
|
|
# include "device/memory.h"
|
|
# include "device/queue.h"
|
|
|
|
# include "device/hip/util.h"
|
|
|
|
CCL_NAMESPACE_BEGIN
|
|
|
|
class HIPDevice;
|
|
class device_memory;
|
|
|
|
/* Base class for HIP queues. */
|
|
class HIPDeviceQueue : public DeviceQueue {
|
|
public:
|
|
HIPDeviceQueue(HIPDevice *device);
|
|
~HIPDeviceQueue() override;
|
|
|
|
int num_concurrent_states(const size_t state_size) const override;
|
|
int num_concurrent_busy_states(const size_t state_size) const override;
|
|
|
|
void init_execution() override;
|
|
|
|
bool enqueue(DeviceKernel kernel,
|
|
const int work_size,
|
|
const DeviceKernelArguments &args) override;
|
|
|
|
bool synchronize() override;
|
|
|
|
void zero_to_device(device_memory &mem) override;
|
|
void copy_to_device(device_memory &mem) override;
|
|
void copy_from_device(device_memory &mem) override;
|
|
|
|
virtual hipStream_t stream()
|
|
{
|
|
return hip_stream_;
|
|
}
|
|
|
|
unique_ptr<DeviceGraphicsInterop> graphics_interop_create() override;
|
|
|
|
protected:
|
|
HIPDevice *hip_device_;
|
|
hipStream_t hip_stream_;
|
|
|
|
void assert_success(hipError_t result, const char *operation);
|
|
};
|
|
|
|
CCL_NAMESPACE_END
|
|
|
|
#endif /* WITH_HIP */
|