Merge pull request #3005 from afonso360/aarch64-i128-extend

aarch64: Implement uextend/sextend for  i128 values

cranelift/filetests/filetests/isa/aarch64/extend-op.clif

@@ -16,3 +16,109 @@ block0(v0: i8):
 ; nextln:  add x0, x1, x0, SXTB
 ; nextln:  ldp fp, lr, [sp], #16
 ; nextln:  ret
+
+
+function %i128_uextend_i64(i64) -> i128 {
+block0(v0: i64):
+    v1 = uextend.i128 v0
+    return v1
+}
+
+; check: stp fp, lr, [sp, #-16]!
+; nextln: mov fp, sp
+; nextln: movz x1, #0
+; nextln: ldp fp, lr, [sp], #16
+; nextln: ret
+
+function %i128_sextend_i64(i64) -> i128 {
+block0(v0: i64):
+    v1 = sextend.i128 v0
+    return v1
+}
+
+; check: stp fp, lr, [sp, #-16]!
+; nextln: mov fp, sp
+; nextln: asr x1, x0, #63
+; nextln: ldp fp, lr, [sp], #16
+; nextln: ret
+
+
+function %i128_uextend_i32(i32) -> i128 {
+block0(v0: i32):
+    v1 = uextend.i128 v0
+    return v1
+}
+
+; check: stp fp, lr, [sp, #-16]!
+; nextln: mov fp, sp
+; nextln: mov w0, w0
+; nextln: movz x1, #0
+; nextln: ldp fp, lr, [sp], #16
+; nextln: ret
+
+function %i128_sextend_i32(i32) -> i128 {
+block0(v0: i32):
+    v1 = sextend.i128 v0
+    return v1
+}
+
+; check: stp fp, lr, [sp, #-16]!
+; nextln: mov fp, sp
+; nextln: sxtw x0, w0
+; nextln: asr x1, x0, #63
+; nextln: ldp fp, lr, [sp], #16
+; nextln: ret
+
+
+function %i128_uextend_i16(i16) -> i128 {
+block0(v0: i16):
+    v1 = uextend.i128 v0
+    return v1
+}
+
+; check: stp fp, lr, [sp, #-16]!
+; nextln: mov fp, sp
+; nextln: uxth w0, w0
+; nextln: movz x1, #0
+; nextln: ldp fp, lr, [sp], #16
+; nextln: ret
+
+function %i128_sextend_i16(i16) -> i128 {
+block0(v0: i16):
+    v1 = sextend.i128 v0
+    return v1
+}
+
+; check: stp fp, lr, [sp, #-16]!
+; nextln: mov fp, sp
+; nextln: sxth x0, w0
+; nextln: asr x1, x0, #63
+; nextln: ldp fp, lr, [sp], #16
+; nextln: ret
+
+
+function %i128_uextend_i8(i8) -> i128 {
+block0(v0: i8):
+    v1 = uextend.i128 v0
+    return v1
+}
+
+; check: stp fp, lr, [sp, #-16]!
+; nextln: mov fp, sp
+; nextln: uxtb w0, w0
+; nextln: movz x1, #0
+; nextln: ldp fp, lr, [sp], #16
+; nextln: ret
+
+function %i128_sextend_i8(i8) -> i128 {
+block0(v0: i8):
+    v1 = sextend.i128 v0
+    return v1
+}
+
+; check: stp fp, lr, [sp, #-16]!
+; nextln: mov fp, sp
+; nextln: sxtb x0, w0
+; nextln: asr x1, x0, #63
+; nextln: ldp fp, lr, [sp], #16
+; nextln: ret

cranelift/filetests/filetests/runtests/i128-extend-2.clif

@@ -0,0 +1,69 @@
+test run
+target aarch64
+target x86_64 machinst
+; TODO:  Merge this file with i128-extend once the x86 legacy backend is removed
+
+function %i128_uextend_i32(i32) -> i64, i64 {
+block0(v0: i32):
+    v1 = uextend.i128 v0
+    v2, v3 = isplit v1
+    return v2, v3
+}
+; run: %i128_uextend_i32(0) == [0, 0]
+; run: %i128_uextend_i32(-1) == [0xffff_ffff, 0]
+; run: %i128_uextend_i32(0xffff_eeee) == [0xffff_eeee, 0]
+
+function %i128_sextend_i32(i32) -> i64, i64 {
+block0(v0: i32):
+    v1 = sextend.i128 v0
+    v2, v3 = isplit v1
+    return v2, v3
+}
+; run: %i128_sextend_i32(0) == [0, 0]
+; run: %i128_sextend_i32(-1) == [-1, -1]
+; run: %i128_sextend_i32(0x7fff_ffff) == [0x7fff_ffff, 0x0000_0000_0000_0000]
+; run: %i128_sextend_i32(0xffff_eeee) == [0xffff_ffff_ffff_eeee, 0xffff_ffff_ffff_ffff]
+
+
+function %i128_uextend_i16(i16) -> i64, i64 {
+block0(v0: i16):
+    v1 = uextend.i128 v0
+    v2, v3 = isplit v1
+    return v2, v3
+}
+; run: %i128_uextend_i16(0) == [0, 0]
+; run: %i128_uextend_i16(-1) == [0xffff, 0]
+; run: %i128_uextend_i16(0xffee) == [0xffee, 0]
+
+function %i128_sextend_i16(i16) -> i64, i64 {
+block0(v0: i16):
+    v1 = sextend.i128 v0
+    v2, v3 = isplit v1
+    return v2, v3
+}
+; run: %i128_sextend_i16(0) == [0, 0]
+; run: %i128_sextend_i16(-1) == [-1, -1]
+; run: %i128_sextend_i16(0x7fff) == [0x7fff, 0x0000_0000_0000_0000]
+; run: %i128_sextend_i16(0xffee) == [0xffff_ffff_ffff_ffee, 0xffff_ffff_ffff_ffff]
+
+
+function %i128_uextend_i8(i8) -> i64, i64 {
+block0(v0: i8):
+    v1 = uextend.i128 v0
+    v2, v3 = isplit v1
+    return v2, v3
+}
+; run: %i128_uextend_i8(0) == [0, 0]
+; run: %i128_uextend_i8(-1) == [0xff, 0]
+; run: %i128_uextend_i8(0xfe) == [0xfe, 0]
+
+function %i128_sextend_i8(i8) -> i64, i64 {
+block0(v0: i8):
+    v1 = sextend.i128 v0
+    v2, v3 = isplit v1
+    return v2, v3
+}
+; run: %i128_sextend_i8(0) == [0, 0]
+; run: %i128_sextend_i8(-1) == [-1, -1]
+; run: %i128_sextend_i8(0x7f) == [0x7f, 0x0000_0000_0000_0000]
+; run: %i128_sextend_i8(0xfe) == [0xffff_ffff_ffff_fffe, 0xffff_ffff_ffff_ffff]

cranelift/filetests/filetests/runtests/i128-extend.clif

@@ -1,29 +1,26 @@
 test run
-; target aarch64 TODO: Not yet implemented on aarch64
+target aarch64
 ; target s390x TODO: Not yet implemented on s390x
 target x86_64 machinst
 target x86_64 legacy
 
-function %i128_uextend() -> b1 {
-block0:
-    v0 = iconst.i64 0xffff_ffff_eeee_0000
+function %i128_uextend_i64(i64) -> i64, i64 {
+block0(v0: i64):
     v1 = uextend.i128 v0
     v2, v3 = isplit v1
-    v4 = icmp_imm eq v2, 0xffff_ffff_eeee_0000
-    v5 = icmp_imm eq v3, 0
-    v6 = band v4, v5
-    return v6
+    return v2, v3
 }
-; run
+; run: %i128_uextend_i64(0) == [0, 0]
+; run: %i128_uextend_i64(-1) == [-1, 0]
+; run: %i128_uextend_i64(0xffff_ffff_eeee_0000) == [0xffff_ffff_eeee_0000, 0]
 
-function %i128_sextend() -> b1 {
-block0:
-    v0 = iconst.i64 0xffff_ffff_eeee_0000
+function %i128_sextend_i64(i64) -> i64, i64 {
+block0(v0: i64):
     v1 = sextend.i128 v0
     v2, v3 = isplit v1
-    v4 = icmp_imm eq v2, 0xffff_ffff_eeee_0000
-    v5 = icmp_imm eq v3, 0xffff_ffff_ffff_ffff
-    v6 = band v4, v5
-    return v6
+    return v2, v3
 }
-; run
+; run: %i128_sextend_i64(0) == [0, 0]
+; run: %i128_sextend_i64(-1) == [-1, -1]
+; run: %i128_sextend_i64(0x7fff_ffff_ffff_ffff) == [0x7fff_ffff_ffff_ffff, 0x0000_0000_0000_0000]
+; run: %i128_sextend_i64(0xffff_ffff_eeee_0000) == [0xffff_ffff_eeee_0000, 0xffff_ffff_ffff_ffff]