1 files changed, 4 insertions, 4 deletions

diff --git a/include/linux/sched.h b/include/linux/sched.h
index 26a2e6122734..18f197223ebd 100644
--- a/include/linux/sched.h
+++ b/include/linux/sched.h
@@ -252,7 +252,7 @@ extern char ___assert_task_state[1 - 2*!!(
 #define set_task_state(tsk, state_value)                        \
         do {                                                    \
                 (tsk)->task_state_change = _THIS_IP_;           \
-                set_mb((tsk)->state, (state_value));            \
+                smp_store_mb((tsk)->state, (state_value));              \
         } while (0)
 
 /*
@@ -274,7 +274,7 @@ extern char ___assert_task_state[1 - 2*!!(
 #define set_current_state(state_value)                          \
         do {                                                    \
                 current->task_state_change = _THIS_IP_;         \
-                set_mb(current->state, (state_value));          \
+                smp_store_mb(current->state, (state_value));            \
         } while (0)
 
 #else
@@ -282,7 +282,7 @@ extern char ___assert_task_state[1 - 2*!!(
 #define __set_task_state(tsk, state_value)              \
         do { (tsk)->state = (state_value); } while (0)
 #define set_task_state(tsk, state_value)                \
-        set_mb((tsk)->state, (state_value))
+        smp_store_mb((tsk)->state, (state_value))
 
 /*
  * set_current_state() includes a barrier so that the write of current->state
@@ -298,7 +298,7 @@ extern char ___assert_task_state[1 - 2*!!(
 #define __set_current_state(state_value)                \
         do { current->state = (state_value); } while (0)
 #define set_current_state(state_value)                  \
-        set_mb(current->state, (state_value))
+        smp_store_mb(current->state, (state_value))
 
 #endif