1 files changed, 17 insertions, 2 deletions

diff --git a/nvdebug_entry.c b/nvdebug_entry.c
index 0abe658..84e8a6a 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;
+                              *num_gpcs_entry, *lce_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();
@@ -261,8 +261,23 @@ int __init nvdebug_init(void) {
                         (void*)NV_FUSE_GPC);
                 // In both nouveau and nvgpu, the PCE_MAP register is only available on Volta+
                 if (g_nvdebug_state[res].chip_id >= NV_CHIP_ID_VOLTA) {
+                        
+                        //create a pce mask for iteration
+                        u32 ce_pce_map = nvdebug_readl(&g_nvdebug_state[device_id], NV_CE_PCE_MAP);
+                        u32 num_pce =  U32(hweight32(ce_pce_map));
+                        u32 disabled_pce_mask = ~ce_pce_map;
+                        char file_name[20];
+                        for (int pce_id = 0; pce_id < num_pce; pce_id++){
+                                //if pce is disabled, do nothing
+                                if ((1 << pce_id) & disabled_pce_mask)
+                                        continue;
+                                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*)NV_LCE_FOR_PCE(pce_id));                                 
+                        }
+
                         // TODO: Redo to num_pces
-                        // Create file `/proc/gpu#/pce_map`, world readable
                         num_gpcs_entry = proc_create_data(
                                 "pce_map", 0444, dir, compat_ops(&nvdebug_read_reg32_file_ops),
                                 (void*)NV_CE_PCE_MAP);