Differential D15902 Diff 57407 intern/cycles/device/optix/device_impl.h

Changeset View

Standalone View

intern/cycles/device/optix/device_impl.h

/* SPDX-License-Identifier: Apache-2.0		/* SPDX-License-Identifier: Apache-2.0
* Copyright 2019, NVIDIA Corporation.		* Copyright 2019, NVIDIA Corporation.
* Copyright 2019-2022 Blender Foundation. */		* Copyright 2019-2022 Blender Foundation. */

#pragma once		#pragma once

#ifdef WITH_OPTIX		#ifdef WITH_OPTIX

# include "device/cuda/device_impl.h"		# include "device/cuda/device_impl.h"
# include "device/optix/queue.h"		# include "device/optix/queue.h"
# include "device/optix/util.h"		# include "device/optix/util.h"
		# include "kernel/osl/globals.h"
# include "kernel/types.h"		# include "kernel/types.h"
# include "util/unique_ptr.h"		# include "util/unique_ptr.h"

CCL_NAMESPACE_BEGIN		CCL_NAMESPACE_BEGIN

class BVHOptiX;		class BVHOptiX;
struct KernelParamsOptiX;		struct KernelParamsOptiX;

/* List of OptiX program groups. */		/* List of OptiX program groups. */
enum {		enum {
PG_RGEN_INTERSECT_CLOSEST,		PG_RGEN_INTERSECT_CLOSEST,
PG_RGEN_INTERSECT_SHADOW,		PG_RGEN_INTERSECT_SHADOW,
PG_RGEN_INTERSECT_SUBSURFACE,		PG_RGEN_INTERSECT_SUBSURFACE,
PG_RGEN_INTERSECT_VOLUME_STACK,		PG_RGEN_INTERSECT_VOLUME_STACK,
		PG_RGEN_SHADE_BACKGROUND,
		PG_RGEN_SHADE_LIGHT,
		PG_RGEN_SHADE_SURFACE,
PG_RGEN_SHADE_SURFACE_RAYTRACE,		PG_RGEN_SHADE_SURFACE_RAYTRACE,
PG_RGEN_SHADE_SURFACE_MNEE,		PG_RGEN_SHADE_SURFACE_MNEE,
		PG_RGEN_SHADE_VOLUME,
		PG_RGEN_SHADE_SHADOW,
		PG_RGEN_EVAL_DISPLACE,
		PG_RGEN_EVAL_BACKGROUND,
		PG_RGEN_EVAL_CURVE_SHADOW_TRANSPARENCY,
PG_MISS,		PG_MISS,
PG_HITD, /* Default hit group. */		PG_HITD, /* Default hit group. */
PG_HITS, /* __SHADOW_RECORD_ALL__ hit group. */		PG_HITS, /* __SHADOW_RECORD_ALL__ hit group. */
PG_HITL, /* __BVH_LOCAL__ hit group (only used for triangles). */		PG_HITL, /* __BVH_LOCAL__ hit group (only used for triangles). */
PG_HITV, /* __VOLUME__ hit group. */		PG_HITV, /* __VOLUME__ hit group. */
PG_HITD_MOTION,		PG_HITD_MOTION,
PG_HITS_MOTION,		PG_HITS_MOTION,
PG_HITD_POINTCLOUD,		PG_HITD_POINTCLOUD,
PG_HITS_POINTCLOUD,		PG_HITS_POINTCLOUD,
PG_CALL_SVM_AO,		PG_CALL_SVM_AO,
PG_CALL_SVM_BEVEL,		PG_CALL_SVM_BEVEL,
NUM_PROGRAM_GROUPS		NUM_PROGRAM_GROUPS
};		};

static const int MISS_PROGRAM_GROUP_OFFSET = PG_MISS;		static const int MISS_PROGRAM_GROUP_OFFSET = PG_MISS;
static const int NUM_MIS_PROGRAM_GROUPS = 1;		static const int NUM_MISS_PROGRAM_GROUPS = 1;
static const int HIT_PROGAM_GROUP_OFFSET = PG_HITD;		static const int HIT_PROGAM_GROUP_OFFSET = PG_HITD;
static const int NUM_HIT_PROGRAM_GROUPS = 8;		static const int NUM_HIT_PROGRAM_GROUPS = 8;
static const int CALLABLE_PROGRAM_GROUPS_BASE = PG_CALL_SVM_AO;		static const int CALLABLE_PROGRAM_GROUPS_BASE = PG_CALL_SVM_AO;
static const int NUM_CALLABLE_PROGRAM_GROUPS = 2;		static const int NUM_CALLABLE_PROGRAM_GROUPS = 2;

/* List of OptiX pipelines. */		/* List of OptiX pipelines. */
enum { PIP_SHADE_RAYTRACE, PIP_SHADE_MNEE, PIP_INTERSECT, NUM_PIPELINES };		enum { PIP_SHADE, PIP_INTERSECT, NUM_PIPELINES };

/* A single shader binding table entry. */		/* A single shader binding table entry. */
struct SbtRecord {		struct SbtRecord {
char header[OPTIX_SBT_RECORD_HEADER_SIZE];		char header[OPTIX_SBT_RECORD_HEADER_SIZE];
};		};

class OptiXDevice : public CUDADevice {		class OptiXDevice : public CUDADevice {
public:		public:
OptixDeviceContext context = NULL;		OptixDeviceContext context = NULL;

OptixModule optix_module = NULL; /* All necessary OptiX kernels are in one module. */		OptixModule optix_module = NULL; /* All necessary OptiX kernels are in one module. */
OptixModule builtin_modules[2] = {};		OptixModule builtin_modules[2] = {};
OptixPipeline pipelines[NUM_PIPELINES] = {};		OptixPipeline pipelines[NUM_PIPELINES] = {};
		OptixProgramGroup groups[NUM_PROGRAM_GROUPS] = {};
		OptixPipelineCompileOptions pipeline_options = {};

bool motion_blur = false;
device_vector<SbtRecord> sbt_data;		device_vector<SbtRecord> sbt_data;
device_only_memory<KernelParamsOptiX> launch_params;		device_only_memory<KernelParamsOptiX> launch_params;
OptixTraversableHandle tlas_handle = 0;

		# 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;		vector<unique_ptr<device_only_memory<char>>> delayed_free_bvh_memory;
thread_mutex delayed_free_bvh_mutex;		thread_mutex delayed_free_bvh_mutex;

class Denoiser {		class Denoiser {
public:		public:
explicit Denoiser(OptiXDevice *device);		explicit Denoiser(OptiXDevice *device);

OptiXDevice *device;		OptiXDevice *device;
Show All 17 Lines	public:
bool use_pass_flow = false;		bool use_pass_flow = false;
};		};
Denoiser denoiser_;		Denoiser denoiser_;

public:		public:
OptiXDevice(const DeviceInfo &info, Stats &stats, Profiler &profiler);		OptiXDevice(const DeviceInfo &info, Stats &stats, Profiler &profiler);
~OptiXDevice();		~OptiXDevice();

private:
BVHLayoutMask get_bvh_layout_mask() const override;		BVHLayoutMask get_bvh_layout_mask() const override;

string compile_kernel_get_common_cflags(const uint kernel_features);		string compile_kernel_get_common_cflags(const uint kernel_features);

bool load_kernels(const uint kernel_features) override;		bool load_kernels(const uint kernel_features) override;

		bool load_osl_kernels() override;

bool build_optix_bvh(BVHOptiX *bvh,		bool build_optix_bvh(BVHOptiX *bvh,
OptixBuildOperation operation,		OptixBuildOperation operation,
const OptixBuildInput &build_input,		const OptixBuildInput &build_input,
uint16_t num_motion_steps);		uint16_t num_motion_steps);

void build_bvh(BVH *bvh, Progress &progress, bool refit) override;		void build_bvh(BVH *bvh, Progress &progress, bool refit) override;

void release_optix_bvh(BVH *bvh) override;		void release_optix_bvh(BVH *bvh) override;
void free_bvh_memory_delayed();		void free_bvh_memory_delayed();

void const_copy_to(const char name, void host, size_t size) override;		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);		void update_launch_params(size_t offset, void *data, size_t data_size);

virtual unique_ptr<DeviceQueue> gpu_queue_create() override;		virtual unique_ptr<DeviceQueue> gpu_queue_create() override;

		void *get_cpu_osl_memory() override;

/* --------------------------------------------------------------------		/* --------------------------------------------------------------------
* Denoising.		* Denoising.
*/		*/

class DenoiseContext;		class DenoiseContext;
class DenoisePass;		class DenoisePass;

virtual bool denoise_buffer(const DeviceDenoiseTask &task) override;		virtual bool denoise_buffer(const DeviceDenoiseTask &task) override;
▲ Show 20 Lines • Show All 43 Lines • Show Last 20 Lines