Engage!

1 files changed, 23 insertions, 0 deletions

diff --git a/llama.cpp/ggml/src/ggml-virtgpu/virtgpu-shm.h b/llama.cpp/ggml/src/ggml-virtgpu/virtgpu-shm.h
new file mode 100644
index 0000000..606860a
--- /dev/null
+++ b/llama.cpp/ggml/src/ggml-virtgpu/virtgpu-shm.h
@@ -0,0 +1,23 @@
+#pragma once
+
+#include "virtgpu-utils.h"
+
+#include <sys/mman.h>
+
+#include <atomic>
+#include <cassert>
+#include <cstddef>
+#include <cstdint>
+
+struct virtgpu;
+
+struct virtgpu_shmem {
+    uint32_t res_id;
+    size_t   mmap_size;
+    void *   mmap_ptr;
+
+    uint32_t gem_handle;
+};
+
+int  virtgpu_shmem_create(virtgpu * gpu, size_t size, virtgpu_shmem * shmem);
+void virtgpu_shmem_destroy(virtgpu * gpu, virtgpu_shmem * shmem);