Fix blas lock bug on AArch64.

This commit is contained in:
Zhang Xianyi 2015-06-26 11:54:41 +08:00
parent d8f18d32c3
commit 3f1b57668e
1 changed files with 9 additions and 8 deletions

View File

@ -46,21 +46,22 @@ USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
static void __inline blas_lock(volatile BLASULONG *address){
int register ret;
int register tmp;
do {
while (*address) {YIELDING;};
__asm__ __volatile__(
"1: \n\t"
"ldrex r2, [%1] \n\t"
"mov r2, #0 \n\t"
"strex r3, r2, [%1] \n\t"
"cmp r3, #0 \n\t"
"bne 1b \n\t"
"mov %0 , r3 \n\t"
: "=r"(ret), "=r"(address)
"ldaxr %2, [%1] \n\t"
"mov %2, #0 \n\t"
"stlxr %w0, %2, [%1] \n\t"
"cbnz %w0, 1b \n\t"
"mov %0 , #0 \n\t"
: "=r"(ret), "=r"(address), "=r"(tmp)
: "1"(address)
: "memory", "r2" , "r3"
: "memory", "%w0"
//, "%r2" , "%r3"
);