#include <asm/mmu_context.h>
 #include <asm/facility.h>
 
-static struct static_key have_mvcos = STATIC_KEY_INIT_FALSE;
+static DEFINE_STATIC_KEY_FALSE(have_mvcos);
 
 static inline unsigned long copy_from_user_mvcos(void *x, const void __user *ptr,
                                                 unsigned long size)
 
 unsigned long __copy_from_user(void *to, const void __user *from, unsigned long n)
 {
-       if (static_key_false(&have_mvcos))
+       if (static_branch_likely(&have_mvcos))
                return copy_from_user_mvcos(to, from, n);
        return copy_from_user_mvcp(to, from, n);
 }
 
 unsigned long __copy_to_user(void __user *to, const void *from, unsigned long n)
 {
-       if (static_key_false(&have_mvcos))
+       if (static_branch_likely(&have_mvcos))
                return copy_to_user_mvcos(to, from, n);
        return copy_to_user_mvcs(to, from, n);
 }
 
 unsigned long __copy_in_user(void __user *to, const void __user *from, unsigned long n)
 {
-       if (static_key_false(&have_mvcos))
+       if (static_branch_likely(&have_mvcos))
                return copy_in_user_mvcos(to, from, n);
        return copy_in_user_mvc(to, from, n);
 }
 
 unsigned long __clear_user(void __user *to, unsigned long size)
 {
-       if (static_key_false(&have_mvcos))
+       if (static_branch_likely(&have_mvcos))
                        return clear_user_mvcos(to, size);
        return clear_user_xc(to, size);
 }
 static int __init uaccess_init(void)
 {
        if (!uaccess_primary && test_facility(27))
-               static_key_slow_inc(&have_mvcos);
+               static_branch_enable(&have_mvcos);
        return 0;
 }
 early_initcall(uaccess_init);