1 files changed, 2 insertions, 1 deletions

diff --git a/include/litmus/nvidia_info.h b/include/litmus/nvidia_info.h
index 579301d77cf5..9e07a27fdee3 100644
--- a/include/litmus/nvidia_info.h
+++ b/include/litmus/nvidia_info.h
@@ -7,7 +7,8 @@
 #include <litmus/litmus_softirq.h>
 
 
-#define NV_DEVICE_NUM NR_LITMUS_SOFTIRQD
+//#define NV_DEVICE_NUM NR_LITMUS_SOFTIRQD
+#define NV_DEVICE_NUM CONFIG_NV_DEVICE_NUM
 
 int init_nvidia_info(void);