545.23.06

src/nvidia/inc/kernel/gpu/gpu_shared_data_map.h

@@ -1,5 +1,5 @@
 /*
- * SPDX-FileCopyrightText: Copyright (c) 2022 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+ * SPDX-FileCopyrightText: Copyright (c) 2022-2023 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
  * SPDX-License-Identifier: MIT
  *
  * Permission is hereby granted, free of charge, to any person obtaining a
@@ -28,22 +28,29 @@
 #include "gpu/mem_mgr/mem_desc.h"
 #include "class/cl00de.h"
 
+#include "tmr.h"
+
 // ****************************************************************************
 //                          Type definitions
 // ****************************************************************************
+typedef struct RusdQueryCache RUSD_QUERY_CACHE;
 
 typedef struct GpuSharedDataMap {
     MEMORY_DESCRIPTOR *pMemDesc;
     NvP64 pMapBuffer;
     NvP64 pMapBufferPriv;
+    NvU64 lastPolledDataMask;
     NvU32 processId;
 
-    NV00DE_SHARED_DATA data;
-} GpuSharedDataMap;
+    TMR_EVENT *pRusdRefreshTmrEvent;
 
-typedef struct GspUserSharedData {
-    NvU32 gspAssertCount;
-} GspUserSharedData;
+    NV00DE_SHARED_DATA data;
+
+    // Private data to assist metrics query
+    RUSD_QUERY_CACHE *pRusdQueryCache;
+    NvU8 curGroup;
+    NvBool bWorkItemPending;
+} GpuSharedDataMap;
 
 /**
  * Start data write, returns data struct to write into
@@ -52,8 +59,12 @@ typedef struct GspUserSharedData {
  * call gpushareddataWriteFinish to push the new data into the user mapping
  */
 NV00DE_SHARED_DATA * gpushareddataWriteStart(OBJGPU *pGpu);
+
 // Finish data write, pushes data cached by above into mapped data
 void gpushareddataWriteFinish(OBJGPU *pGpu);
 
+// Unmap and free memory in GpuSharedDataMap
+void gpushareddataDestroy(OBJGPU *pGpu);
+
 #endif /* GPU_SHARED_DATA_MAP_H */