Make rcu_thread_key context-aware