aboutsummaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
-rw-r--r--include/linux/percpu-refcount.h12
-rw-r--r--lib/percpu-refcount.c2
2 files changed, 7 insertions, 7 deletions
diff --git a/include/linux/percpu-refcount.h b/include/linux/percpu-refcount.h
index dd2a08600453..95961f0bf62d 100644
--- a/include/linux/percpu-refcount.h
+++ b/include/linux/percpu-refcount.h
@@ -105,7 +105,7 @@ static inline void percpu_ref_get(struct percpu_ref *ref)
105{ 105{
106 unsigned __percpu *pcpu_count; 106 unsigned __percpu *pcpu_count;
107 107
108 rcu_read_lock(); 108 rcu_read_lock_sched();
109 109
110 pcpu_count = ACCESS_ONCE(ref->pcpu_count); 110 pcpu_count = ACCESS_ONCE(ref->pcpu_count);
111 111
@@ -114,7 +114,7 @@ static inline void percpu_ref_get(struct percpu_ref *ref)
114 else 114 else
115 atomic_inc(&ref->count); 115 atomic_inc(&ref->count);
116 116
117 rcu_read_unlock(); 117 rcu_read_unlock_sched();
118} 118}
119 119
120/** 120/**
@@ -134,7 +134,7 @@ static inline bool percpu_ref_tryget(struct percpu_ref *ref)
134 unsigned __percpu *pcpu_count; 134 unsigned __percpu *pcpu_count;
135 int ret = false; 135 int ret = false;
136 136
137 rcu_read_lock(); 137 rcu_read_lock_sched();
138 138
139 pcpu_count = ACCESS_ONCE(ref->pcpu_count); 139 pcpu_count = ACCESS_ONCE(ref->pcpu_count);
140 140
@@ -143,7 +143,7 @@ static inline bool percpu_ref_tryget(struct percpu_ref *ref)
143 ret = true; 143 ret = true;
144 } 144 }
145 145
146 rcu_read_unlock(); 146 rcu_read_unlock_sched();
147 147
148 return ret; 148 return ret;
149} 149}
@@ -159,7 +159,7 @@ static inline void percpu_ref_put(struct percpu_ref *ref)
159{ 159{
160 unsigned __percpu *pcpu_count; 160 unsigned __percpu *pcpu_count;
161 161
162 rcu_read_lock(); 162 rcu_read_lock_sched();
163 163
164 pcpu_count = ACCESS_ONCE(ref->pcpu_count); 164 pcpu_count = ACCESS_ONCE(ref->pcpu_count);
165 165
@@ -168,7 +168,7 @@ static inline void percpu_ref_put(struct percpu_ref *ref)
168 else if (unlikely(atomic_dec_and_test(&ref->count))) 168 else if (unlikely(atomic_dec_and_test(&ref->count)))
169 ref->release(ref); 169 ref->release(ref);
170 170
171 rcu_read_unlock(); 171 rcu_read_unlock_sched();
172} 172}
173 173
174#endif 174#endif
diff --git a/lib/percpu-refcount.c b/lib/percpu-refcount.c
index 8bf9e719cca0..7deeb6297a48 100644
--- a/lib/percpu-refcount.c
+++ b/lib/percpu-refcount.c
@@ -154,5 +154,5 @@ void percpu_ref_kill_and_confirm(struct percpu_ref *ref,
154 (((unsigned long) ref->pcpu_count)|PCPU_REF_DEAD); 154 (((unsigned long) ref->pcpu_count)|PCPU_REF_DEAD);
155 ref->confirm_kill = confirm_kill; 155 ref->confirm_kill = confirm_kill;
156 156
157 call_rcu(&ref->rcu, percpu_ref_kill_rcu); 157 call_rcu_sched(&ref->rcu, percpu_ref_kill_rcu);
158} 158}