1 files changed, 97 insertions, 0 deletions
diff --git a/intern/cycles/device/metal/queue.h b/intern/cycles/device/metal/queue.h
new file mode 100644
index 00000000000..7aafcb2efe4
--- /dev/null
+++ b/intern/cycles/device/metal/queue.h
@@ -0,0 +1,97 @@
+/*
+ * 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
+
+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 */