1 files changed, 3 insertions, 0 deletions

diff --git a/include/litmus/nvidia_info.h b/include/litmus/nvidia_info.h
index f28d8f0a7f40..c5bf3dc0a210 100644
--- a/include/litmus/nvidia_info.h
+++ b/include/litmus/nvidia_info.h
@@ -9,5 +9,8 @@ int is_nvidia_func(void *func);
 
 int is_nvidia_in_callstack(void);
 
+// Returns the Nvidia device # associated with the
+// provided tasklet.
+u32 get_nv_device_num(const struct tasklet_struct *t);
 
 #endif