1 files changed, 10 insertions, 3 deletions

diff --git a/nvdebug_entry.c b/nvdebug_entry.c
index e77b053..c444ff7 100644
--- a/nvdebug_entry.c
+++ b/nvdebug_entry.c
@@ -203,7 +203,7 @@ int create_tpc_mask_files(int device_id, struct proc_dir_entry *dir) {
 int __init nvdebug_init(void) {
         struct proc_dir_entry *dir, *preempt_entry, *disable_channel_entry,
                               *enable_channel_entry, *switch_to_tsg_entry, *device_info_entry,
-                              *num_gpcs_entry, *lce_for_pce_entry;
+                              *num_gpcs_entry, *lce_for_pce_entry, *grce_for_pce_entry;
         int rl_create_err, tpc_masks_create_err;
         // Check that an NVIDIA GPU is present and initialize g_nvdebug_state
         int res = probe_and_cache_device();
@@ -275,8 +275,15 @@ int __init nvdebug_init(void) {
                                 snprintf(file_name, 20, "lce_for_pce%d",pce_id);
                                 lce_for_pce_entry = proc_create_data(
                                 file_name, 0444, dir, compat_ops(&nvdebug_read_reg32_file_ops),
-                                (void*)(uintptr_t)NV_LCE_FOR_PCE(pce_id));                                      
-                        }
+                                (void*)(uintptr_t)NV_LCE_FOR_PCE(pce_id));
+                                snprintf(file_name, 20, "grce_for_pce%d",pce_id);
+                                grce_for_pce_entry = proc_create_data(
+                                file_name, 0444, dir, compat_ops(&nvdebug_read_reg32_file_ops),
+                                (void*)(uintptr_t)NV_GRCE_FOR_CE(pce_id));
+                                                
+                                if (!lce_for_pce_entry || !grce_for_pce_entry)
+                                        return -ENOMEM; 
+                        }               
 
                         // TODO: Redo to num_pces
                         num_gpcs_entry = proc_create_data(