[S390] Improve code generated by atomic operations.