1 files changed, 4 insertions, 0 deletions

diff --git a/userspace/include/unit/unit.h b/userspace/include/unit/unit.h
index 2f93bab5..442a73f1 100644
--- a/userspace/include/unit/unit.h
+++ b/userspace/include/unit/unit.h
@@ -23,6 +23,8 @@
 #ifndef __UNIT_UNIT_H__
 #define __UNIT_UNIT_H__
 
+#include <pthread.h>
+
 struct gk20a;
 
 struct unit_module;
@@ -84,6 +86,8 @@ struct unit_module {
          */
         void                            *lib_handle;
         struct unit_fw                  *fw;
+
+        pthread_t thread;
 };
 
 /*