| Scott McMurray | 3b16c04 | 2022-10-24 00:02:50 | [diff] [blame] | 1 | // compile-flags: -O |
| 2 | // min-llvm-version: 15.0 (LLVM 13 in CI does this differently from submodule LLVM) |
| Scott McMurray | 9d4b1f9 | 2022-11-16 23:58:43 | [diff] [blame] | 3 | // ignore-debug (because unchecked is checked in debug) |
| Scott McMurray | 3b16c04 | 2022-10-24 00:02:50 | [diff] [blame] | 4 | |
| 5 | #![crate_type = "lib"] |
| 6 | #![feature(unchecked_math)] |
| 7 | |
| 8 | // CHECK-LABEL: @unchecked_shl_unsigned_same |
| 9 | #[no_mangle] |
| 10 | pub unsafe fn unchecked_shl_unsigned_same(a: u32, b: u32) -> u32 { |
| 11 | // CHECK-NOT: and i32 |
| 12 | // CHECK: shl i32 %a, %b |
| 13 | // CHECK-NOT: and i32 |
| 14 | a.unchecked_shl(b) |
| 15 | } |
| 16 | |
| 17 | // CHECK-LABEL: @unchecked_shl_unsigned_smaller |
| 18 | #[no_mangle] |
| 19 | pub unsafe fn unchecked_shl_unsigned_smaller(a: u16, b: u32) -> u16 { |
| 20 | // This uses -DAG to avoid failing on irrelevant reorderings, |
| 21 | // like emitting the truncation earlier. |
| 22 | |
| 23 | // CHECK-DAG: %[[INRANGE:.+]] = icmp ult i32 %b, 65536 |
| 24 | // CHECK-DAG: tail call void @llvm.assume(i1 %[[INRANGE]]) |
| 25 | // CHECK-DAG: %[[TRUNC:.+]] = trunc i32 %b to i16 |
| 26 | // CHECK-DAG: shl i16 %a, %[[TRUNC]] |
| 27 | a.unchecked_shl(b) |
| 28 | } |
| 29 | |
| 30 | // CHECK-LABEL: @unchecked_shl_unsigned_bigger |
| 31 | #[no_mangle] |
| 32 | pub unsafe fn unchecked_shl_unsigned_bigger(a: u64, b: u32) -> u64 { |
| 33 | // CHECK: %[[EXT:.+]] = zext i32 %b to i64 |
| 34 | // CHECK: shl i64 %a, %[[EXT]] |
| 35 | a.unchecked_shl(b) |
| 36 | } |
| 37 | |
| 38 | // CHECK-LABEL: @unchecked_shr_signed_same |
| 39 | #[no_mangle] |
| 40 | pub unsafe fn unchecked_shr_signed_same(a: i32, b: u32) -> i32 { |
| 41 | // CHECK-NOT: and i32 |
| 42 | // CHECK: ashr i32 %a, %b |
| 43 | // CHECK-NOT: and i32 |
| 44 | a.unchecked_shr(b) |
| 45 | } |
| 46 | |
| 47 | // CHECK-LABEL: @unchecked_shr_signed_smaller |
| 48 | #[no_mangle] |
| 49 | pub unsafe fn unchecked_shr_signed_smaller(a: i16, b: u32) -> i16 { |
| 50 | // This uses -DAG to avoid failing on irrelevant reorderings, |
| 51 | // like emitting the truncation earlier. |
| 52 | |
| 53 | // CHECK-DAG: %[[INRANGE:.+]] = icmp ult i32 %b, 32768 |
| 54 | // CHECK-DAG: tail call void @llvm.assume(i1 %[[INRANGE]]) |
| 55 | // CHECK-DAG: %[[TRUNC:.+]] = trunc i32 %b to i16 |
| 56 | // CHECK-DAG: ashr i16 %a, %[[TRUNC]] |
| 57 | a.unchecked_shr(b) |
| 58 | } |
| 59 | |
| 60 | // CHECK-LABEL: @unchecked_shr_signed_bigger |
| 61 | #[no_mangle] |
| 62 | pub unsafe fn unchecked_shr_signed_bigger(a: i64, b: u32) -> i64 { |
| 63 | // CHECK: %[[EXT:.+]] = zext i32 %b to i64 |
| 64 | // CHECK: ashr i64 %a, %[[EXT]] |
| 65 | a.unchecked_shr(b) |
| 66 | } |
