2 files changed, 5 insertions, 0 deletions

diff --git a/userspace/include/unit/args.h b/userspace/include/unit/args.h
index def84a29..708a1655 100644
--- a/userspace/include/unit/args.h
+++ b/userspace/include/unit/args.h
@@ -42,6 +42,7 @@ struct unit_fw_args {
         bool             help;
         int              verbose_lvl;
         bool             no_color;
+        int              thread_count;
 
         const char      *unit_name;
         const char      *unit_load_path;
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;
 };
 
 /*