1 files changed, 2 insertions, 0 deletions

diff --git a/nvdebug_entry.c b/nvdebug_entry.c
index 0cf5344..68e4d71 100644
--- a/nvdebug_entry.c
+++ b/nvdebug_entry.c
@@ -109,6 +109,7 @@ int probe_and_cache_devices(void) {
                 g_nvdebug_state[i].chip_id = ids.chip_id;
                 g_nvdebug_state[i].pcid = NULL;
                 g_nvdebug_state[i].bar3 = NULL;
+                g_nvdebug_state[i].dev = dev;
                 printk(KERN_INFO "[nvdebug] Chip ID %x (architecture %s) detected on platform bus and initialized.",
                        ids.chip_id, ARCH2NAME(ids.architecture));
                 i++;
@@ -131,6 +132,7 @@ int probe_and_cache_devices(void) {
                 if (!g_nvdebug_state[i].bar3)
                         g_nvdebug_state[i].bar3 = pci_iomap(pcid, 3, pci_resource_len(pcid, 3)/2);
                 g_nvdebug_state[i].pcid = pcid;
+                g_nvdebug_state[i].dev = &pcid->dev;
                 ids.raw = nvdebug_readl(&g_nvdebug_state[i], NV_MC_BOOT_0);
                 if (ids.raw == -1) {
                         pci_err(pcid, "[nvdebug] Unable to read config from Master Controller on this GPU\n");