4d7bd22beb
* Add GraphicsInteropDevice to check if interop is possible with device * Rename GraphcisInterop to GraphicsInteropBuffer * Include display device type and memory size in GraphicsInteropBuffer * Unnest graphics interop class to make forward declarations possible Pull Request: https://projects.blender.org/blender/blender/pulls/137363
115 lines
3.3 KiB
C++
115 lines
3.3 KiB
C++
/* SPDX-FileCopyrightText: 2011-2022 Blender Foundation
|
|
*
|
|
* SPDX-License-Identifier: Apache-2.0 */
|
|
|
|
#pragma once
|
|
|
|
#ifdef WITH_HIP
|
|
|
|
# include "device/device.h"
|
|
# include "device/hip/kernel.h"
|
|
# include "device/hip/queue.h"
|
|
# include "device/hip/util.h"
|
|
|
|
# ifdef WITH_HIP_DYNLOAD
|
|
# include "hipew.h"
|
|
# endif
|
|
|
|
CCL_NAMESPACE_BEGIN
|
|
|
|
class DeviceQueue;
|
|
|
|
class HIPDevice : public GPUDevice {
|
|
|
|
friend class HIPContextScope;
|
|
|
|
public:
|
|
hipDevice_t hipDevice;
|
|
hipCtx_t hipContext;
|
|
hipModule_t hipModule;
|
|
int pitch_alignment;
|
|
int hipDevId;
|
|
int hipDevArchitecture;
|
|
int hipRuntimeVersion;
|
|
bool first_error;
|
|
|
|
HIPDeviceKernels kernels;
|
|
|
|
static bool have_precompiled_kernels();
|
|
|
|
BVHLayoutMask get_bvh_layout_mask(uint /*kernel_features*/) const override;
|
|
|
|
void set_error(const string &error) override;
|
|
|
|
HIPDevice(const DeviceInfo &info, Stats &stats, Profiler &profiler, bool headless);
|
|
|
|
~HIPDevice() override;
|
|
|
|
bool support_device(const uint /*kernel_features*/);
|
|
|
|
bool check_peer_access(Device *peer_device) override;
|
|
|
|
bool use_adaptive_compilation();
|
|
|
|
virtual string compile_kernel_get_common_cflags(const uint kernel_features);
|
|
|
|
virtual string compile_kernel(const uint kernel_features,
|
|
const char *name,
|
|
const char *base = "hip");
|
|
|
|
bool load_kernels(const uint kernel_features) override;
|
|
void reserve_local_memory(const uint kernel_features);
|
|
|
|
/* All memory types. */
|
|
void mem_alloc(device_memory &mem) override;
|
|
void mem_copy_to(device_memory &mem) override;
|
|
void mem_move_to_host(device_memory &mem) override;
|
|
void mem_copy_from(
|
|
device_memory &mem, const size_t y, size_t w, const size_t h, size_t elem) override;
|
|
void mem_zero(device_memory &mem) override;
|
|
void mem_free(device_memory &mem) override;
|
|
|
|
device_ptr mem_alloc_sub_ptr(device_memory &mem, const size_t offset, size_t /*size*/) override;
|
|
|
|
/* Global memory. */
|
|
void global_alloc(device_memory &mem);
|
|
void global_copy_to(device_memory &mem);
|
|
void global_free(device_memory &mem);
|
|
|
|
/* Texture memory. */
|
|
void tex_alloc(device_texture &mem);
|
|
void tex_copy_to(device_texture &mem);
|
|
void tex_free(device_texture &mem);
|
|
|
|
/* Device side memory. */
|
|
void get_device_memory_info(size_t &total, size_t &free) override;
|
|
bool alloc_device(void *&device_pointer, const size_t size) override;
|
|
void free_device(void *device_pointer) override;
|
|
|
|
/* Shared memory. */
|
|
bool shared_alloc(void *&shared_pointer, const size_t size) override;
|
|
void shared_free(void *shared_pointer) override;
|
|
void *shared_to_device_pointer(const void *shared_pointer) override;
|
|
|
|
/* Memory copy. */
|
|
void copy_host_to_device(void *device_pointer, void *host_pointer, const size_t size) override;
|
|
void const_copy_to(const char *name, void *host, const size_t size) override;
|
|
|
|
/* Graphics resources interoperability. */
|
|
bool should_use_graphics_interop(const GraphicsInteropDevice &interop_device,
|
|
const bool log) override;
|
|
|
|
unique_ptr<DeviceQueue> gpu_queue_create() override;
|
|
|
|
int get_num_multiprocessors();
|
|
int get_max_num_threads_per_multiprocessor();
|
|
|
|
protected:
|
|
bool get_device_attribute(hipDeviceAttribute_t attribute, int *value);
|
|
int get_device_default_attribute(hipDeviceAttribute_t attribute, const int default_value);
|
|
};
|
|
|
|
CCL_NAMESPACE_END
|
|
|
|
#endif
|