b803d7fabb
Since #118841 there are more cases where Cycles would check for the graphics interop support. This could lead to a crash when graphics interop functions are called without having active graphics context. This change makes it so there is no graphics interop calls when doing headless render. In order to achieve this the device creation is now aware of the headless mode. Pull Request: https://projects.blender.org/blender/blender/pulls/122844
112 lines
3.0 KiB
C++
112 lines
3.0 KiB
C++
/* SPDX-FileCopyrightText: 2011-2022 Blender Foundation
|
|
*
|
|
* SPDX-License-Identifier: Apache-2.0 */
|
|
|
|
#ifdef WITH_CUDA
|
|
|
|
# include "device/cuda/kernel.h"
|
|
# include "device/cuda/queue.h"
|
|
# include "device/cuda/util.h"
|
|
# include "device/device.h"
|
|
|
|
# include "util/map.h"
|
|
|
|
# ifdef WITH_CUDA_DYNLOAD
|
|
# include "cuew.h"
|
|
# else
|
|
# include <cuda.h>
|
|
# include <cudaGL.h>
|
|
# endif
|
|
|
|
CCL_NAMESPACE_BEGIN
|
|
|
|
class DeviceQueue;
|
|
|
|
class CUDADevice : public GPUDevice {
|
|
|
|
friend class CUDAContextScope;
|
|
|
|
public:
|
|
CUdevice cuDevice;
|
|
CUcontext cuContext;
|
|
CUmodule cuModule;
|
|
int pitch_alignment;
|
|
int cuDevId;
|
|
int cuDevArchitecture;
|
|
bool first_error;
|
|
|
|
CUDADeviceKernels kernels;
|
|
|
|
static bool have_precompiled_kernels();
|
|
|
|
virtual BVHLayoutMask get_bvh_layout_mask(uint /*kernel_features*/) const override;
|
|
|
|
void set_error(const string &error) override;
|
|
|
|
CUDADevice(const DeviceInfo &info, Stats &stats, Profiler &profiler, bool headless);
|
|
|
|
virtual ~CUDADevice();
|
|
|
|
bool support_device(const uint /*kernel_features*/);
|
|
|
|
bool check_peer_access(Device *peer_device) override;
|
|
|
|
bool use_adaptive_compilation();
|
|
|
|
string compile_kernel_get_common_cflags(const uint kernel_features);
|
|
|
|
string compile_kernel(const string &cflags,
|
|
const char *name,
|
|
const char *base = "cuda",
|
|
bool force_ptx = false);
|
|
|
|
virtual bool load_kernels(const uint kernel_features) override;
|
|
|
|
void reserve_local_memory(const uint kernel_features);
|
|
|
|
virtual void get_device_memory_info(size_t &total, size_t &free) override;
|
|
virtual bool alloc_device(void *&device_pointer, size_t size) override;
|
|
virtual void free_device(void *device_pointer) override;
|
|
virtual bool alloc_host(void *&shared_pointer, size_t size) override;
|
|
virtual void free_host(void *shared_pointer) override;
|
|
virtual void transform_host_pointer(void *&device_pointer, void *&shared_pointer) override;
|
|
virtual void copy_host_to_device(void *device_pointer, void *host_pointer, size_t size) override;
|
|
|
|
void mem_alloc(device_memory &mem) override;
|
|
|
|
void mem_copy_to(device_memory &mem) override;
|
|
|
|
void mem_copy_from(device_memory &mem, size_t y, size_t w, 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, size_t offset, size_t /*size*/) override;
|
|
|
|
virtual void const_copy_to(const char *name, void *host, size_t size) override;
|
|
|
|
void global_alloc(device_memory &mem);
|
|
|
|
void global_free(device_memory &mem);
|
|
|
|
void tex_alloc(device_texture &mem);
|
|
|
|
void tex_free(device_texture &mem);
|
|
|
|
virtual bool should_use_graphics_interop() override;
|
|
|
|
virtual unique_ptr<DeviceQueue> gpu_queue_create() override;
|
|
|
|
int get_num_multiprocessors();
|
|
int get_max_num_threads_per_multiprocessor();
|
|
|
|
protected:
|
|
bool get_device_attribute(CUdevice_attribute attribute, int *value);
|
|
int get_device_default_attribute(CUdevice_attribute attribute, int default_value);
|
|
};
|
|
|
|
CCL_NAMESPACE_END
|
|
|
|
#endif
|