vllm-ascend vnpu v1

csrc/idle_offload/shm_worker.h

@@ -0,0 +1,29 @@
+#pragma once
+
+#include "shm_helper.h"
+
+class ShmWorker {
+ public:
+  ShmWorker();
+  ~ShmWorker();
+
+  bool register_worker(int32_t tgid, uint64_t *out_shareable_handle,
+                       uint64_t *out_vmem_size);
+
+  bool lock_gpu();
+  void unlock_gpu();
+
+  
+ private:
+  int32_t tgid;
+  ShmHelper *shm_helper;
+  std::thread heart_beat_thread;
+  std::atomic<bool> stop_heart_beat;
+
+  // request
+  uint64_t make_request(uint32_t type, uint64_t parameter);
+  int register_worker_shm();
+
+  // heart beat
+  void heart_beat_loop(int slot);
+};