1 files changed, 9 insertions, 1 deletions

diff --git a/nvdebug.h b/nvdebug.h
index 8d78135..b0e6bb8 100644
--- a/nvdebug.h
+++ b/nvdebug.h
@@ -689,12 +689,20 @@ typedef union {
   Also see dev_ce.ref.txt of NVIDIA's open-gpu-doc for info.
 */
 #define NV_PTOP_SCAL_NUM_CES 0x00022444
+// Defined number of GRCEs for a GPU
+# define NV_GRCE_NUM 2
 // Defined GRCE->CE mapping offsets from nvgpu
-#define NV_GRCE_FOR_CE_GV100(i) (0x00104034+(i)*4)
+#define NV_GRCE_FOR_CE(i) (0x00104034+(i)*4)
 // Defined LCE->PCE mapping offset from nvgpu (same as ce_pce2lce_config_r(i) in nvgpu)
 #define NV_LCE_FOR_PCE_GV100(i) (0x00104040+(i)*4)
 #define NV_LCE_FOR_PCE_GA100(i) (0x00104100+(i)*4)
 #define NV_LCE_FOR_PCE_GP100(i) (0x0010402c+(i)/2)
+#define NV_LCE_FOR_PCE_TU104(i) (0x00104040+(i)*4)
+// Defined struct for storing PCE index and offset for proc_create
+struct combo {
+        uint32_t offset:32;
+        uint32_t index:32;
+};
 
 /* Physical Copy Engine (PCE) information
   On Pascal GPUs or newer, this register complements the above information by