1 files changed, 2 insertions, 0 deletions

diff --git a/lib/locking-selftest.c b/lib/locking-selftest.c
index 1e1bbf171eca..a1705545e6ac 100644
--- a/lib/locking-selftest.c
+++ b/lib/locking-selftest.c
@@ -1989,6 +1989,7 @@ void locking_selftest(void)
 
         init_shared_classes();
         debug_locks_silent = !debug_locks_verbose;
+        lockdep_set_selftest_task(current);
 
         DO_TESTCASE_6R("A-A deadlock", AA);
         DO_TESTCASE_6R("A-B-B-A deadlock", ABBA);
@@ -2097,5 +2098,6 @@ void locking_selftest(void)
                 printk("---------------------------------\n");
                 debug_locks = 1;
         }
+        lockdep_set_selftest_task(NULL);
         debug_locks_silent = 0;
 }