patched issues with GPU compatability for CE_MAP

1 files changed, 9 insertions, 4 deletions

diff --git a/nvdebug.h b/nvdebug.h
index 213a786..8d78135 100644
--- a/nvdebug.h
+++ b/nvdebug.h
@@ -389,7 +389,9 @@ typedef union {
 #define NV_CHIP_ID_GP106 0x136 // Discrete GeForce GTX 1060
 #define NV_CHIP_ID_GV11B 0x15B // Jetson Xavier embedded GPU
 #define NV_CHIP_ID_KEPLER 0x0E0
+#define NV_CHIP_ID_PASCAL 0x130
 #define NV_CHIP_ID_VOLTA 0x140
+#define NV_CHIP_ID_TURING 0x160
 #define NV_CHIP_ID_AMPERE 0x170
 
 inline static const char* ARCH2NAME(uint32_t arch) {
@@ -687,10 +689,12 @@ typedef union {
   Also see dev_ce.ref.txt of NVIDIA's open-gpu-doc for info.
 */
 #define NV_PTOP_SCAL_NUM_CES 0x00022444
-//defined GRCE->CE mapping offset from nvgpu
-#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(i)(0x00104040+(i)*4)
+// Defined GRCE->CE mapping offsets from nvgpu
+#define NV_GRCE_FOR_CE_GV100(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)
 
 /* Physical Copy Engine (PCE) information
   On Pascal GPUs or newer, this register complements the above information by
@@ -704,6 +708,7 @@ typedef union {
   Also see dev_ce.ref.txt of NVIDIA's open-gpu-doc for info.
 */
 #define NV_CE_PCE_MAP 0x00104028
+#define MAP_SIZE 32
 
 
 /* Location of the 1Kb instance block with page tables for BAR1 and BAR2.