Differential D13423 Diff 45772 intern/cycles/device/metal/queue.h

Changeset View

Standalone View

intern/cycles/device/metal/queue.h

This file was added.

				/*
				* Copyright 2021 Blender Foundation
				*
				* Licensed under the Apache License, Version 2.0 (the "License");
				* you may not use this file except in compliance with the License.
				* You may obtain a copy of the License at
				*
				* http://www.apache.org/licenses/LICENSE-2.0
				*
				* Unless required by applicable law or agreed to in writing, software
				* distributed under the License is distributed on an "AS IS" BASIS,
				* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
				* See the License for the specific language governing permissions and
				* limitations under the License.
				*/

				#pragma once

				#ifdef WITH_METAL

				# include "device/kernel.h"
				# include "device/memory.h"
				# include "device/queue.h"

				# include "device/metal/util.h"
				# include "kernel/device/metal/globals.h"

				# define metal_printf VLOG(4) << string_printf

				brechtUnsubmitted Done Inline Actions For debugging logs, you can use `VLOG(4)`. brecht: For debugging logs, you can use `VLOG(4)`.
				michael_jonesAuthorUnsubmitted Done Inline Actions Not sure if this is what you had in mind, or whether you meant to use VLOG(4) directly instead of metal_printf, but having a specialised metal_printf to hand is useful for isolating and intercepting logging. michael_jones: Not sure if this is what you had in mind, or whether you meant to use VLOG(4) directly instead…
				brechtUnsubmitted Done Inline Actions This is fine. The main thing is not to print debugging info to the console by default, which this accomplishes. brecht: This is fine. The main thing is not to print debugging info to the console by default, which…
				CCL_NAMESPACE_BEGIN

				class MetalDevice;

				/* Base class for Metal queues. */
				class MetalDeviceQueue : public DeviceQueue {
				public:
				MetalDeviceQueue(MetalDevice *device);
				~MetalDeviceQueue();

				virtual int num_concurrent_states(const size_t) const override;
				virtual int num_concurrent_busy_states() const override;

				virtual void init_execution() override;

				virtual bool enqueue(DeviceKernel kernel,
				const int work_size,
				DeviceKernelArguments const &args) override;

				virtual bool synchronize() override;

				virtual void zero_to_device(device_memory &mem) override;
				virtual void copy_to_device(device_memory &mem) override;
				virtual void copy_from_device(device_memory &mem) override;

				virtual bool kernel_available(DeviceKernel kernel) const override;

				protected:
				void prepare_resources(DeviceKernel kernel);

				id<MTLComputeCommandEncoder> get_compute_encoder(DeviceKernel kernel);
				id<MTLBlitCommandEncoder> get_blit_encoder();

				MetalDevice *metal_device;
				MetalBufferPool temp_buffer_pool;

				API_AVAILABLE(macos(11.0), ios(14.0))
				MTLCommandBufferDescriptor *command_buffer_desc = nullptr;
				id<MTLDevice> mtlDevice = nil;
				id<MTLCommandQueue> mtlCommandQueue = nil;
				id<MTLCommandBuffer> mtlCommandBuffer = nil;
				id<MTLComputeCommandEncoder> mtlComputeEncoder = nil;
				id<MTLBlitCommandEncoder> mtlBlitEncoder = nil;
				id<MTLSharedEvent> shared_event = nil;
				MTLSharedEventListener *shared_event_listener = nil;

				dispatch_queue_t event_queue;
				dispatch_semaphore_t wait_semaphore;

				struct CopyBack {
				void *host_pointer;
				void *gpu_mem;
				uint64_t size;
				};
				std::vector<CopyBack> copy_back_mem;

				uint64_t shared_event_id;
				uint64_t command_buffers_submitted = 0;
				uint64_t command_buffers_completed = 0;
				Stats &stats;

				void close_compute_encoder();
				void close_blit_encoder();
				};

				CCL_NAMESPACE_END

				#endif /* WITH_METAL */