projects / rustc.git / blame
| Commit | Line | Data |
|---|---|---|
| 487cf647 FG |
1 | // compile-flags: -O |
| 2 | // min-llvm-version: 15.0 (LLVM 13 in CI does this differently from submodule LLVM) | |
| 3 | // ignore-debug (because unchecked is checked in debug) | |
| 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 | } |