wsl/hsakmt: add mem flag to protect queue cmd buffer

Signed-off-by: Flora Cui <flora.cui@amd.com>

inc/wddm/gpu_memory.h

@@ -117,6 +117,7 @@ struct GpuMemoryDesc {
       uint32_t is_external : 1;
       uint32_t is_physical_only : 1;
       uint32_t is_locked : 1;
+      uint32_t is_queue_referenced : 1;
 
       uint32_t unused : 27;
     };
@@ -167,6 +168,9 @@ public:
     return (desc_.adapter_luid.HighPart == luid.HighPart &&
       desc_.adapter_luid.LowPart == luid.LowPart);
   }
+  inline void GetQueueReference() { desc_.flags.is_queue_referenced = 1; }
+  inline void PutQueueReference() { desc_.flags.is_queue_referenced = 0; }
+  inline bool IsQueueReferenced() const { return desc_.flags.is_queue_referenced; }
 
   WinAllocationHandle GetAllocationHandle(size_t index) const { return alloc_handles_ptr_[index]; }
   size_t NumChunks() const { return num_allocations_; }

inc/wddm/queue.h

@@ -87,6 +87,7 @@ public:
   virtual hsa_status_t Init(void) { return HSA_STATUS_SUCCESS; }
   virtual hsa_status_t Fini(void) { return HSA_STATUS_SUCCESS; }
   virtual void RingDoorbell() { }
+  virtual void* GetHsaQueueAddr(void) const { return reinterpret_cast<void*>(GetCmdbufAddr()); }
 
   hsa_status_t SwsInit(void);
   hsa_status_t SwsFini(void);
@@ -162,6 +163,7 @@ public:
 
   uint64_t GetAqlWriteIndex(void) const { return cmdbuf_aql_frame_write_index; }
   uint32_t GetAqlFrameSize(void) const { return cmdbuf_aql_frame_size; }
+  void* GetHsaQueueAddr(void) const { return ring; }
 
   bool IsInvalidPacket(void) const {
     uint16_t *packet = (uint16_t *)((char *)ring +
@@ -278,6 +280,7 @@ public:
   uint64_t * GetRingRptr(void) { return WDDMQueue::GetSyncAddr(); }
   uint64_t * GetDoorbellPtr() { return &doorbell_; }
   void RingDoorbell();
+  void* GetHsaQueueAddr(void) const { return reinterpret_cast<void*>(GetCmdbufAddr()); }
 
 private:
   uint64_t wptr_next_;