1 files changed, 4 insertions, 1 deletions

diff --git a/include/asm-s390/processor.h b/include/asm-s390/processor.h
index 8bd14de69e35..4ec652ebb3b1 100644
--- a/include/asm-s390/processor.h
+++ b/include/asm-s390/processor.h
@@ -203,7 +203,10 @@ unsigned long get_wchan(struct task_struct *p);
 # define cpu_relax()    asm volatile ("diag 0,0,68" : : : "memory")
 #else /* __s390x__ */
 # define cpu_relax() \
-        asm volatile ("ex 0,%0" : : "i" (__LC_DIAG44_OPCODE) : "memory")
+        do { \
+                if (MACHINE_HAS_DIAG44) \
+                        asm volatile ("diag 0,0,68" : : : "memory"); \
+        } while (0)
 #endif /* __s390x__ */
 
 /*