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
124 lines
3.4 KiB
C++
124 lines
3.4 KiB
C++
/* SPDX-FileCopyrightText: 2019 NVIDIA Corporation
|
|
* SPDX-FileCopyrightText: 2019-2022 Blender Foundation
|
|
*
|
|
* SPDX-License-Identifier: Apache-2.0 */
|
|
|
|
#pragma once
|
|
|
|
#ifdef WITH_OPTIX
|
|
|
|
# include "device/cuda/device_impl.h"
|
|
# include "device/optix/util.h"
|
|
# include "kernel/osl/globals.h"
|
|
|
|
CCL_NAMESPACE_BEGIN
|
|
|
|
class BVHOptiX;
|
|
struct KernelParamsOptiX;
|
|
|
|
/* List of OptiX program groups. */
|
|
enum {
|
|
PG_RGEN_INTERSECT_CLOSEST,
|
|
PG_RGEN_INTERSECT_SHADOW,
|
|
PG_RGEN_INTERSECT_SUBSURFACE,
|
|
PG_RGEN_INTERSECT_VOLUME_STACK,
|
|
PG_RGEN_INTERSECT_DEDICATED_LIGHT,
|
|
PG_RGEN_SHADE_BACKGROUND,
|
|
PG_RGEN_SHADE_LIGHT,
|
|
PG_RGEN_SHADE_SURFACE,
|
|
PG_RGEN_SHADE_SURFACE_RAYTRACE,
|
|
PG_RGEN_SHADE_SURFACE_MNEE,
|
|
PG_RGEN_SHADE_VOLUME,
|
|
PG_RGEN_SHADE_SHADOW,
|
|
PG_RGEN_SHADE_DEDICATED_LIGHT,
|
|
PG_RGEN_EVAL_DISPLACE,
|
|
PG_RGEN_EVAL_BACKGROUND,
|
|
PG_RGEN_EVAL_CURVE_SHADOW_TRANSPARENCY,
|
|
PG_MISS,
|
|
PG_HITD, /* Default hit group. */
|
|
PG_HITS, /* __SHADOW_RECORD_ALL__ hit group. */
|
|
PG_HITL, /* __BVH_LOCAL__ hit group (only used for triangles). */
|
|
PG_HITV, /* __VOLUME__ hit group. */
|
|
PG_HITD_MOTION,
|
|
PG_HITS_MOTION,
|
|
PG_HITD_POINTCLOUD,
|
|
PG_HITS_POINTCLOUD,
|
|
PG_CALL_SVM_AO,
|
|
PG_CALL_SVM_BEVEL,
|
|
NUM_PROGRAM_GROUPS
|
|
};
|
|
|
|
static const int MISS_PROGRAM_GROUP_OFFSET = PG_MISS;
|
|
static const int NUM_MISS_PROGRAM_GROUPS = 1;
|
|
static const int HIT_PROGAM_GROUP_OFFSET = PG_HITD;
|
|
static const int NUM_HIT_PROGRAM_GROUPS = 8;
|
|
static const int CALLABLE_PROGRAM_GROUPS_BASE = PG_CALL_SVM_AO;
|
|
static const int NUM_CALLABLE_PROGRAM_GROUPS = 2;
|
|
|
|
/* List of OptiX pipelines. */
|
|
enum { PIP_SHADE, PIP_INTERSECT, NUM_PIPELINES };
|
|
|
|
/* A single shader binding table entry. */
|
|
struct SbtRecord {
|
|
char header[OPTIX_SBT_RECORD_HEADER_SIZE];
|
|
};
|
|
|
|
class OptiXDevice : public CUDADevice {
|
|
public:
|
|
OptixDeviceContext context = NULL;
|
|
|
|
OptixModule optix_module = NULL; /* All necessary OptiX kernels are in one module. */
|
|
OptixModule builtin_modules[2] = {};
|
|
OptixPipeline pipelines[NUM_PIPELINES] = {};
|
|
OptixProgramGroup groups[NUM_PROGRAM_GROUPS] = {};
|
|
OptixPipelineCompileOptions pipeline_options = {};
|
|
|
|
device_vector<SbtRecord> sbt_data;
|
|
device_only_memory<KernelParamsOptiX> launch_params;
|
|
|
|
# ifdef WITH_OSL
|
|
OSLGlobals osl_globals;
|
|
vector<OptixModule> osl_modules;
|
|
vector<OptixProgramGroup> osl_groups;
|
|
# endif
|
|
|
|
private:
|
|
OptixTraversableHandle tlas_handle = 0;
|
|
vector<unique_ptr<device_only_memory<char>>> delayed_free_bvh_memory;
|
|
thread_mutex delayed_free_bvh_mutex;
|
|
|
|
public:
|
|
OptiXDevice(const DeviceInfo &info, Stats &stats, Profiler &profiler, bool headless);
|
|
~OptiXDevice();
|
|
|
|
BVHLayoutMask get_bvh_layout_mask(uint /*kernel_features*/) const override;
|
|
|
|
string compile_kernel_get_common_cflags(const uint kernel_features);
|
|
|
|
bool load_kernels(const uint kernel_features) override;
|
|
|
|
bool load_osl_kernels() override;
|
|
|
|
bool build_optix_bvh(BVHOptiX *bvh,
|
|
OptixBuildOperation operation,
|
|
const OptixBuildInput &build_input,
|
|
uint16_t num_motion_steps);
|
|
|
|
void build_bvh(BVH *bvh, Progress &progress, bool refit) override;
|
|
|
|
void release_bvh(BVH *bvh) override;
|
|
void free_bvh_memory_delayed();
|
|
|
|
void const_copy_to(const char *name, void *host, size_t size) override;
|
|
|
|
void update_launch_params(size_t offset, void *data, size_t data_size);
|
|
|
|
virtual unique_ptr<DeviceQueue> gpu_queue_create() override;
|
|
|
|
void *get_cpu_osl_memory() override;
|
|
};
|
|
|
|
CCL_NAMESPACE_END
|
|
|
|
#endif /* WITH_OPTIX */
|