1 files changed, 5 insertions, 4 deletions

diff --git a/nvdebug_entry.c b/nvdebug_entry.c
index c362209..0abe658 100644
--- a/nvdebug_entry.c
+++ b/nvdebug_entry.c
@@ -146,7 +146,7 @@ int probe_and_cache_device(void) {
 
 // Create files `/proc/gpu#/runlist#`, world readable
 int create_runlist_files(int device_id, struct proc_dir_entry *dir) {
-        ptop_device_info_t info;
+        ptop_device_info_gk104_t info;
         struct proc_dir_entry *rl_entry;
         int i, rl_id;
         char runlist_name[12];
@@ -154,8 +154,8 @@ int create_runlist_files(int device_id, struct proc_dir_entry *dir) {
         // Figure out how many runlists there are by checking the device info
         // registers. Runlists are always numbered sequentially, so we just have
         // to find the highest-valued one and add 1 to get the number of runlists.
-        for (i = 0; i < NV_PTOP_DEVICE_INFO__SIZE_1; i++) {
-                info.raw = nvdebug_readl(&g_nvdebug_state[device_id], NV_PTOP_DEVICE_INFO(i));
+        for (i = 0; i < NV_PTOP_DEVICE_INFO__SIZE_1_GK104; i++) {
+                info.raw = nvdebug_readl(&g_nvdebug_state[device_id], NV_PTOP_DEVICE_INFO_GK104(i));
                 if (info.info_type != INFO_TYPE_ENUM || !info.runlist_is_valid)
                         continue;
                 if (info.runlist_enum > max_rl_id)
@@ -219,7 +219,8 @@ int __init nvdebug_init(void) {
                 if (!(dir = proc_mkdir_data(device_id_str, 0555, NULL, (void*)device_id)))
                         goto out_nomem;
                 // Create files `/proc/gpu#/runlist#`, world readable
-                rl_create_err = create_runlist_files(device_id, dir);
+                if (g_nvdebug_state[device_id].chip_id < NV_CHIP_ID_AMPERE)
+                        create_runlist_files(device_id, dir);
                 // Create files `/proc/gpu#/gpc#_tpc_mask`, world readable
                 tpc_masks_create_err = create_tpc_mask_files(device_id, dir);
                 // Create file `/proc/gpu#/preempt_tsg`, world writable